equivalent python de la fonction matlab find

Bonjour à tous,

J'essaie actuellement de traduire mon code matlab de simulation de cycle d'hystérésis en python pour une meilleure performance et pour le faire tourner en mpi.

Ce code contient une ligne dans laquelle j'extrais les indices des cases d'un vecteur (ou d'une liste) qui valident la condition que j'impose. Cette fonction en matlab est la fonction find, voici la ligne de code:

Code:

---

Change = find((Hysteron(2,:) == 1 & Htot <= -Hysteron(4,:)) | (Hysteron(2,:) == -1 & Htot >= Hysteron(4,:)) | (Hysteron(3,:)== 1000000 & Htot > -Hysteron(4,:) & Htot < Hysteron(4,:) & abs(Hysteron(1,:)-Htot./Hysteron(4,:)) > 0.0001));

---

Ici la fonction find renvoie les indices des colonnes de Hysteron (= un tableau à 7 lignes et beaucoup de colonnes) qui satisfont les conditions imposées.

En l’occurrence cette commande est très pratique car elle m'épargne une boucle sur le vecteur Hysteron, qui n'est pas énorme mais qui est balayé plusieurs centaines de fois lors de la simulation. Je précise que ce qui m’intéresse est d'avoir les indices des cases qui valide la condition à l’intérieur de find stockés dans une liste.
Connaissez vous un équivalent python à cette fonction find?

Merci beaucoup.
Bonne journée.

Bonjour,


Je ne connais pas matlab et donc pas la fonction find. Toutefois, si le but est de connaitres les indices qui valent "True" dans un tableau booléen, il y a la fonction "numpy.where". Cela donnerait quelque chose comme ça:

Code:

---

1 2 3 4 5

import numpy as np Change = np.where( (Hysteron[1,:] == 1 & Htot <= -Hysteron[3,:]) | (Hysteron[1,:] == -1 & Htot >= Hysteron[3,:]) | (Hysteron[2,:] == 1000000 & Htot > -Hysteron[3,:] & Htot < Hysteron[3,:] & abs(Hysteron[0,:]-Htot./Hysteron[3,:]) > 0.0001))[0]

---

C'est exactement ce que je cherchais!

Merci beaucoup!

Bonne fin de journée

Impossible de corriger cette bouillie.
Édite ton message et utilise les balises de code (#).
Le sujet a à voir avec le titre ? Sinon, ouvre un nouveau fil.