NumPy数组中连续值的总和 [英] Sum slices of consecutive values in a NumPy array

问题描述

假设我有一个包含10个值的numpy数组a.此处只是一个示例情况，尽管我想对长度为100的数组重复相同的操作.

Let's say I have a numpy array a containing 10 values. Just an example situation here, although I would like to repeat the same for an array with length 100.

a = np.array([1,2,3,4,5,6,7,8,9,10])

我想对前5个值求和，然后对后5个值求和，依此类推，然后将它们存储在一个新的空白列表中，例如b.

I would like to sum the first 5 values followed by the second 5 values and so on and store them in a new empty list say b.

所以b将包含b = [15,40].

我该怎么做?

推荐答案

尝试使用此列表理解:

b = [sum(a[current: current+5]) for current in xrange(0, len(a), 5)]

从列表中一次获取5个切片，将它们相加并构造一个列表.也适用于长度不是5的倍数的列表.

It takes slices of 5 at a time from the list, sums them up and constructs a list. Also works for lists which aren't a multiple of 5 in length.

(xrange在python3 +中应为range)

(xrange should be range in python3+)

这篇关于NumPy数组中连续值的总和的文章就介绍到这了，希望我们推荐的答案对大家有所帮助，也希望大家多多支持IT屋！