У меня есть массив
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?