python - 计算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])