nd-array indexation avec une variable array

Bonjour,

Commençons par le code :

Code:

---

1 2 3 4

import numpy as np m=np.zeros((2,2,4),dtype='bool') i = [1,1]

---

Alors voilà, j'ai une matrice à n dimensions (ici n=3) et une variable i qui a (n-1) élément indiquant quel élément prendre sur chaque dimension.
Comment donc récupérer les données sur la dernière dimension ?
J'ai naïvement essayé m[i], mais niet.

Code:

---

1 2 3

m[1,1] # renvoie bien un array de dimension 4 m[i] # renvoie bêtement la matrice m entière

---

Une idée ?
Merci

Salut,

Code:

---

1 2 3 4 5 6 7 8 9 10 11 12 13 14

>>> import numpy as np >>> m=np.zeros((2,2,4),dtype='bool') >>> m array([[[False, False, False, False], [False, False, False, False]],   [[False, False, False, False], [False, False, False, False]]], dtype=bool) >>> m[1,1] array([False, False, False, False], dtype=bool) >>> i = (1, 1) >>> m[i] array([False, False, False, False], dtype=bool) >>>

---

notez que mon i est "tuple" et non "list".

- W

Yes, il fallait un tuple !
Merci :)