Discussion :

[quedreuxa]

Bonjour,

je débute en python et j'ai une question très simple : existe-t-il une fonction qui permettrait d'obtenir, dans un ndarray, l'index d'une valeur spécifiée ?

par exemple :

>>> tab = [ [3,2,1], [7,8,9], [4,5,6] ]
>>> tab.index(7)
3

il existe déjà 2 fonctions qui le font pour les valeurs maximum et minimum (respectivement argmax() et argmin() ) mais je ne trouve aucune fonction qui le fait pour une valeur spécifiée...

Quelqu'un peut-il m'aider ?
merci

[josmiley]

si tous les sous-ensembles ont la même taille ( ici 3) tu peux faire:

Code :

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

1
2
3
4
tab = [ [3,2,1], [7,8,9], [4,5,6] ]
idx = sum(tab,[]).index(7)
idx = idx/3,idx%3
>>> (1,0)

[quedreuxa]

Génial, ça marche !
Merci beaucoup josmiley

[kango]

bonjour,

si ce sont des array numpy, il y a aussi where:

Code :

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

1
2
3
4
5
6
7
8
9
10
import numpy as np
tab = np.zeros((3,3),dtype=np.int32)
tab[1,2] = 7
tab[2,0] = 7
print tab
>>>array([[0, 0, 0],
>>>       [0, 0, 7],
>>>       [7, 0, 0]])
print np.where(tab==7)
>>> (array([1, 2]), array([2, 0]))