计算python数组中每5个元素的总和

Question

我有一个 python 数组，我想在其中计算每 5 个元素的总和。就我而言，我有一个包含十个元素的数组c。 （实际上它有更多的元素。）

c = [1, 0, 0, 0, 0, 2, 0, 0, 0, 0]

所以最后我想要一个新数组 (c_new)，它应该显示前 5 个元素和后 5 个元素的总和

所以结果应该是那个

1+0+0+0+0 = 1
2+0+0+0+0 = 2

c_new = [1, 2]

感谢您的帮助 马库斯

Answer 1

您可以通过传递要拆分和求和的索引来使用np.add.reduceat：

import numpy as np
c = [1, 0, 0, 0, 0, 2, 0, 0, 0, 0]
np.add.reduceat(c, np.arange(0, len(c), 5))
# array([1, 2])

Answer 2

这是一种方法 -

c = [1, 0, 0, 0, 0, 2, 0, 0, 0, 0]
print [sum(c[i:i+5]) for i in range(0, len(c), 5)]

结果 -

[1, 2]

Answer 3

如果五除以向量的长度并且它是连续的，那么

np.reshape(c, (-1, 5)).sum(axis=-1)

如果它不是连续的，它也可以工作，但它通常效率较低。

基准测试：

def aredat():
    return np.add.reduceat(c, np.arange(0, len(c), 5))

def reshp():
    np.reshape(c, (-1, 5)).sum(axis=-1)

c = np.random.random(10_000_000)

timeit(aredat, number=100)
3.8516048429883085
timeit(reshp, number=100)
3.09542763303034

所以在可能的情况下，reshapeing 似乎要快一些； reduceat 具有优雅处理非五向量的优势。