Как использовать numpy.where с логическими операторами

Я пытаюсь найти индексы всех элементов в массиве, которые больше, чем a, но меньше b. Вероятно, это просто проблема с моим синтаксисом, но это не работает:

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

Как мне это исправить? Или есть лучший способ сделать это?

Спасибо!

Ответы

Ответ 1

Вот два способа:

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

Для первого (заменив and на &), будьте осторожны, чтобы добавить круглые скобки соответствующим образом: & имеет более высокий приоритет, чем операторы сравнения. Вы также можете использовать *, но я бы не рекомендовал его: он взломан и не делает для считываемого кода.

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