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)
