python - 列表中可变大小的 block 的总和，其中其他列表给出了大小

Question

我想给出以下两个 list 的总和:

a = [0,1,2,3,4,5,6,7,8,9]
b = [2,3,5]

结果应该是 b的每个 a元素的总和，例如:

b[0] = 2，因此第一个总和结果应为:sum(a[0:2])

b[1] = 3，因此第二个求和结果应为:sum(a[2:5])

b[2] = 5，因此第三个求和结果应为:sum(a[5:10])

打印结果: 1,9,35

Answer 1

您可以将np.bincount与weights结合使用:

groups = np.repeat(np.arange(len(b)), b)

np.bincount(groups, weights=a)

输出:

array([ 1.,  9., 35.])