Summe indizierter numpy Arrays

Question

Angenommen, ich habe eine Reihe von Indizes:

I=[0 1 2 3 0 3] 

Und Array von Werten:

W=[w0, w1, w2, w3] 

Wie kann ich ein Array X mit vektorisiert numpy Ausdrücke erzeugt wie folgt erstellen:

X = np.zeros(I.max() + 1) 
for i in range(len(I)): 
    X[I[i]] += W[I[i]] 

im obigen Beispiel X=[w0+w0, w1, w2, w3+w3]

Answer 1

mit reproduzierbaren Objekte starten

>>> I = np.r_[0, 1, 2, 3, 0, 3] 
>>> W = np.r_[60, 50, 40, 30, 20, 10] 
>>> M = W[I] 

Dann beachten Sie, dass man hat

>>> W[I] 
array([60, 50, 40, 30, 60, 30]) 
>>> W[W[I]==W] 
array([60, 50, 40, 30]) 
>>> np.unique(W[I], return_counts=True)[1] 
array([2, 1, 1, 2]) 

Schließlich was

---

Lassen Sie uns

>>> W[M==W]*np.unique(M, return_counts=True)[1] # X 
array([120, 50, 40, 60]) 

über das Tun (fast) zu tun, dass mit Streichern

>>> W = np.array(['w0', 'w1', 'w2', 'w3', 'w4', 'w5']) 
>>> M = W[I] 
>>> M 
array(['w0', 'w1', 'w2', 'w3', 'w0', 'w3'], dtype='|S2') 
>>> W[M==W] 
array(['w0', 'w1', 'w2', 'w3'], dtype='|S2') 
>>> np.unique(W[I], return_counts=True)[1] 
array([2, 1, 1, 2]) 

Welche geistig führt zu betrachten

>>> [2*'w0', 'w1', 'w2', 2*'w3']