python - 是否可以根据另一个向量的整数值组合(添加)一个向量的值

Question

我正在尝试根据另一个向量的整数值添加一个向量的浮点值。

例如，如果我有:

import numpy as np
a = np.array([0.1,0.2,0.3,0.4,0.5,0.6,07.3,0.8,0.9,1.,1.2,1.4])
b = np.array([0,0,0,0,0,1,1,1,2,2,2,2]).astype(int)

我想将 a 向量的前 5 个值加在一起(因为 b 的前 5 个值是 0)，后 3 个值加在一起(因为 b 的后 3 个值是 1)等等。所以最后我希望有

c = function(a,b)
c = [0.1+0.2+0.3+0.4+0.5,  0.6+7.3+0.8, 0.9+1.+1.2+1.4]

Answer 1

方法#1:我们可以利用np.bincount以 b 作为 bins 和 a 作为权重数组 -

In [203]: np.bincount(b,a)
Out[203]: array([1.5, 8.7, 4.5])

方法 #2: 另一种利用矩阵乘法 -

In [210]: (b == np.arange(b.max()+1)[:,None]).dot(a)
Out[210]: array([1.5, 8.7, 4.5])