Cómo usar numpy.donde con operadores lógicos

Question

Cómo usar numpy.donde con operadores lógicos

Estoy tratando de encontrar los índices de todos los elementos en una matriz que son mayores que a pero menores que b. Probablemente es solo un problema con mi sintaxis, pero esto no funciona:

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

¿Cómo debo arreglar esto? ¿O hay una mejor manera de hacerlo?

Gracias!

27

python numpy logical-operators where

Author: mcglashan, 2012-11-27

Source

1 answers

score 46 · Accepted Answer

Aquí hay dos maneras:

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]),)

Para la primera (reemplazando and con &), tenga cuidado de agregar paréntesis apropiadamente: & tiene mayor precedencia que los operadores de comparación. También puedes usar *, pero no lo recomendaría: es hackeado y no hace que el código sea legible.

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