Скажем, я хочу сделать элементарную сумму списка массивов numpy:
tosum = [rand(100,100) for n in range(10)]
Я искал лучший способ сделать это. Кажется, что numpy.sum ужасен:
timeit.timeit('sum(array(tosum), axis=0)',
setup='from numpy import sum; from __main__ import tosum, array',
number=10000)
75.02289700508118
timeit.timeit('sum(tosum, axis=0)',
setup='from numpy import sum; from __main__ import tosum',
number=10000)
78.99106407165527
Уменьшение намного быстрее (на порядок почти на два порядка):
timeit.timeit('reduce(add,tosum)',
setup='from numpy import add; from __main__ import tosum',
number=10000)
1.131795883178711
Похоже, что сокращение даже имеет значимое преимущество над суммой не numpy (обратите внимание, что они используются для 1e6 пробегов, а не 1e4 для вышеуказанных времен):
timeit.timeit('reduce(add,tosum)',
setup='from numpy import add; from __main__ import tosum',
number=1000000)
109.98814797401428
timeit.timeit('sum(tosum)',
setup='from __main__ import tosum',
number=1000000)
125.52461504936218
Есть ли другие методы, которые я должен попробовать? Кто-нибудь может объяснить рейтинги?
Edit
numpy.sum определенно быстрее, если сначала список преобразуется в массив numpy:
tosum2 = array(tosum)
timeit.timeit('sum(tosum2, axis=0)',
setup='from numpy import sum; from __main__ import tosum2',
number=10000)
1.1545608043670654
Тем не менее, мне только интересно делать сумму один раз, поэтому превращение массива в массив numpy по-прежнему будет иметь реальное ограничение производительности.