web-dev-qa-db-fra.com

Comment utiliser numpy.where avec des opérateurs logiques

J'essaie de trouver les indices de tous les éléments d'un tableau qui sont supérieurs à a mais inférieurs à b. C'est probablement juste un problème avec ma syntaxe mais cela ne fonctionne pas:

numpy.where((my_array > a) and (my_array < b))

Comment dois-je résoudre ce problème? Ou existe-t-il une meilleure façon de le faire?

Merci!

40
rferdinand

Voici deux façons:

In [1]: my_array = arange(10)

In [2]: where((my_array > 3) & (my_array < 7))
Out[2]: (array([4, 5, 6]),)

In [3]: where(logical_and(my_array > 3, my_array < 7))
Out[3]: (array([4, 5, 6]),)

Pour le premier (en remplaçant and par &), veillez à bien ajouter les parenthèses: & a une priorité plus élevée que les opérateurs de comparaison. Vous pouvez aussi utiliser *, mais je ne le recommanderais pas: c'est hacky et ne rend pas le code lisible.

In [4]: where((my_array > 3) * (my_array < 7))
Out[4]: (array([4, 5, 6]),)
60
Mark Dickinson