2017-10-10 1 views
0

Je voudrais savoir, s'il y a un moyen d'utiliser un tableau pour sélectionner un élément sur un autre tableau?Comment sélectionner un élément dans un tableau multidimensionnel avec un autre tableau en python?

G=np.array([[[0,0,0],[0,0,0],[0,0,0]], 
      [[1,1,1],[1,8,1],[1,1,1]], 
      [[2,2,2],[2,2,2],[2,2,2]]]) 

Disons que je voulais choisir le '8', je normalement juste faire

In [101]: G[1,1,1] 
Out[101]: 8 

ce que je veux faire quelque chose ist comme

In [102]: c=np.array([1,1,1]) 
In [103]: G[c] 
Out[103]: 8 

Je sais, c'est facile et ça ne marche pas. Ist-il un moyen, cela ne

Répondre

0

utiliser simplement tuple()

G[tuple(c)] 
+0

Merci! J'essaye depuis des jours, mais je n'ai pas trouvé ça. – Impeesa

+0

Je suis content d'avoir pu vous aider! Peut-être que vous pouvez accepter la réponse si vous l'avez trouvé utile? – kaidokuuppa