Get numpy array with only certain indices

Question

Is there a numpy function (or algorithm) that only returns the items of an array depending on and and-function of the values of two other boolean arrays.

E.g.

>>> b1 = numpy.array([False, False, True, True , True])
>>> b2 = numpy.array([True , False, True, False, True])
>>> v  = numpy.array([2    , 4    , 6   , 8,     10  ])

Then the function should return:

numpy.array([6, 10])

Because 6 and 10 are the values where both the corresponding b1 and b2 values are True.

Edited according accepted answer below:

>>> v[b1 & b2]
array([ 6, 10])

Sven Marnach · Accepted Answer · 2012-02-17 17:22:20Z

4

v[b1 & b2]

will do the trick.

answered Feb 17, 2012 at 17:22

Sven Marnach

608k123 gold badges966 silver badges865 bronze badges

Sign up to request clarification or add additional context in comments.

1 Comment

Michel Keijzers Over a year ago

Thank you ... this is the second answer of less than 10 characters with a perfect solution for my problem :-)

Collectives™ on Stack Overflow

Get numpy array with only certain indices

1 Answer 1

1 Comment

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

1 Comment

Your Answer

Sign up or log in

Post as a guest

Related