Discussion :

[bobuse]

Bonjour,

Commençons par le code :

Code :

Sélectionner tout - Visualiser dans une fenêtre à part

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 :

Sélectionner tout - Visualiser dans une fenêtre à part

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

[wiztricks]

Salut,

Code :

Sélectionner tout - Visualiser dans une fenêtre à part

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

[bobuse]

Yes, il fallait un tuple !
Merci