python - 逻辑上添加 numpy

Question

我想做如下操作。但它喜欢直方图操作。

maxIndex = 6    
dst  =zeros((1,6))
a    =array([1,2,3,4,7,0,3,4,5,7])
index=array([1,1,1,3,3,4,4,5,5,5])

a的长度== index的长度，

for i in (a.size):
    dst[index[i]] = dst[index[i]] + a[i]

我怎样才能做到更 pythonic。更高效

Answer 1

如果我没理解错的话，我想你是在找numpy.bincount :

dst = numpy.bincount(index, weights=a, minlength=maxIndex)

这给我 array([ 0., 6., 0., 11., 3., 16.]) 作为输出。如果您不想手动计算 maxIndex，您可以在函数调用中省略 minlength 参数，numpy 将返回一个适当大小的数组给你。