Kumulative Summierung eines Numpy-Arrays nach Index
Angenommen, Sie haben ein Array von Werten, die summiert werden müssen
d = [1,1,1,1,1]
und ein zweites Array, das angibt, welche Elemente summiert werden müssen
i = [0,0,1,2,2]
Das Ergebnis wird in einem neuen Array mit der Größe @ gespeichermax(i)+1
. Also zum Beispieli=[0,0,0,0,0]
wäre gleichbedeutend mit dem Summieren aller Elemente vond
und Speichern des Ergebnisses an Position0
eines neuen Arrays der Größe1
.
Ich habe versucht, dies mit @ zu implementier
c = zeros(max(i)+1)
c[i] += d
Der+=
ie @ -Operation fügt jedes Element nur einmal hinzu, wodurch das unerwartete Ergebnis von @ erhalten wir
[1,1,1]
Anstatt vo
[2,1,2]
Wie würde man diese Art der Summierung richtig umsetzen?