Как найти значения в массиве, которые удовлетворяют двум условиям, используя Python

У меня есть массив

a=[1,2,3,4,5,6,7,8,9]

и я хочу найти индексы элемента s, удовлетворяющие двум условиям: i.e.

a>3 and a<8
ans=[3,4,5,6]
a[ans]=[4,5,6,7]

Я могу использовать numpy.nonzero(a>3) или numpy.nonzero(a<8) но нет numpy.nonzero(a>3 and a<8), который дает ошибку:

ValueError: The truth value of an array with more than one element is
ambiguous. Use a.any() or a.all()

Когда я пытаюсь использовать any или all, я получаю ту же ошибку. Можно ли комбинировать два условных теста, чтобы получить ans?

Ответ 1

numpy.nonzero((a > 3) & (a < 8))

& делает элемент логически булевым и.

Ответ 2

Альтернативой (который я использовал) является numpy.logical_and:

choice = numpy.logical_and(np.greater(a, 3), np.less(a, 8))
numpy.extract(choice, a)

Ответ 3

если вы используете массив numpy, вы можете напрямую использовать ' & ' вместо ' and '.

a=array([1,2,3,4,5,6,7,8,9])
a[(a>3) & (a<8)]
ans=array([3,4,5,6])