增量帶有重複索引的Numpy多維數組

Question

我對Increment Numpy array with repeated indices的多維情況感興趣。

我有一個N維數組和一組N個索引數組，我想增加誰的值。索引數組可能有重複的條目。

沒有重複，解決的辦法是

a = arange(24).reshape(2,3,4) 
i = array([0,0,1]) 
j = array([0,1,1]) 
k = array([0,0,3]) 
a[i,j,k] += 1 

隨着重複，（例如：j=array([0,0,2])），我無法做出numpy的遞增重複。

Answer 1

如何：

import numpy as np 
a = np.zeros((2,3,4)) 
i = np.array([0,0,1]) 
j = np.array([0,0,1]) 
k = np.array([0,0,3]) 

ijk = np.vstack((i,j,k)).T 
H,edge = np.histogramdd(ijk,bins=a.shape) 
a += H 

Answer 2

我不知道是否有直接的數組索引一個更簡單的解決方案，但這個工程：

for x,y,z in zip(i,j,k): 
    a[x,y,z] +=1