numpyでインデックスで配列を累積する方法は？

Question

私は配列を持っている：

a = np.array([0,0,0,0,0,0]) 

私は、インデックスが複数回現れることができますが、それぞれの指標の中にいくつかの他のアレイを追加したいです。私は各インデックスのいくつかを取得したい。

a[np.array([1,2,2,1,3])] += np.array([1,1,1,1,1]) 

をしかしであることを得る：私は書く

array([0, 1, 1, 1, 0, 0]) 

しかし、私が望むことは得ることです：

array([0, 2, 2, 1, 0, 0]) 

forループせずにnumpyの中でこれを実装する方法は？

Answer 1

は純粋numpyを使用して、ループのための回避：

np.add.at(a, np.array([1,2,2,1,3]), np.array([1,1,1,1,1])) 

出力：

>>> a = np.array([0,0,0,0,0,0]) 
>>> np.add.at(a, np.array([1,2,2,1,3]), np.array([1,1,1,1,1])) 
>>> a 
array([0, 2, 2, 1, 0, 0]) 

注意してください、これはインプレースの置換を行います。これはあなたが望むものですが、将来の視聴者には望ましくないかもしれません。したがって、ノート：

Answer 2

あなたはいつでも自分自身を繰り返すことができます。ような何か：

for i in [1,2,2,1,3]: 
    a[i] += 1 

Answer 3

私はこれを行うための巧妙なnumpyのベクトル化された方法を知っていません...私が思い付くことができる最高は次のとおりです。

>>> indices = np.array([1,2,2,1,3]) 
>>> values = np.array([1,1,1,1,1]) 
>>> a = np.array([0,0,0,0,0,0]) 
>>> for i, ix in enumerate(indices): 
... a[ix] += values[i] 
... 
>>> a 
array([0, 2, 2, 1, 0, 0]) 

Answer 4

あなたは（は仮定のような何かを行うことができます各インデックスの相関値）がある：

a = np.array([0,0,0,0,0,0]) 
idxs = np.array([1,2,2,1,3]) 
vals = np.array([1,1,1,1,1]) 
for idx, val in zip(idxs,vals): 
    a[idx] += val