如何仅对 numpy 数组的索引列表执行求和,例如,如果我有一个数组 a = [1,2,3,4]
和要求和的索引列表, indices = [0, 2]
我想要一个快速操作来给我答案 4
因为 中索引 0 和索引 2 处的求和值的值a
是 4
最佳答案
可以在用indices
索引后直接使用sum
:
a = np.array([1,2,3,4])
indices = [0, 2]
a[indices].sum()
关于Python numpy 数组对某些索引求和,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/47734392/