Logo Questions Linux Laravel Mysql Ubuntu Git Menu
 

How find values in an array that meet two conditions using Python

Tags:

python

find

numpy

I have an array

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

and I want to find the indices of the element s that meet two conditions i.e.

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

I can use numpy.nonzero(a>3) or numpy.nonzero(a<8) but not numpy.nonzero(a>3 and a<8) which gives the error:

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

When I try to use any or all I get the same error. Is it possible to combine two conditional tests to get the ans?

like image 216
David Avatar asked Jul 14 '10 16:07

David


2 Answers

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

& does an element-wise boolean and.

like image 111
Matthew Flaschen Avatar answered Nov 03 '22 20:11

Matthew Flaschen


An alternative (which I ended up using) is numpy.logical_and:

choice = numpy.logical_and(np.greater(a, 3), np.less(a, 8))
numpy.extract(choice, a)
like image 24
tssch Avatar answered Nov 03 '22 18:11

tssch