Comment utiliser numpy.où avec les opérateurs logiques
J'essaie de trouver les indices de tous les éléments dans 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 est-il une meilleure façon de le faire?
Merci!
27
demandé sur
mcglashan
2012-11-27 21:10:18
1 réponses
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 à ajouter des parenthèses de manière appropriée: &
a une priorité plus élevée que les opérateurs de comparaison. Vous pouvez également 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]),)
46
répondu
Mark Dickinson
2012-11-27 17:19:36