Indexowanie numpy

sieci-neuronowe

(Kacper Kulczak) #1
p = np.array([[1,2,3,4,5,6], [10,20,30,40,50,60], [100,200,300,400,500,600]])
g = np.array([3,2,4])
p[g] # żeby wyszło coś takiego ---> [3, 20, 4000]

Chciałbym na wyjściu [ p[0][g[0]], p[1][g[1]], p[2][g[2]] ]
Jak to zrobić indeksowaniem w numpy? (będę tym obracał sporymi macierzami)


(Filip Pacanowski) #2

p[np.arange(p.shape[0]), g-1]

https://colab.research.google.com/drive/13ahVSjf2hvZuUxIdvhGYI0OwDQN7BVf0


(Kacper Kulczak) #3

Thx :slight_smile: