Numpy: get only the array rows that have a value in an interval

Question

I have a numpy array which which has a shape of (10, 3) and looks like:

10 | xxx | yyy
11 | xxx | yyy
13 | xxx | yyy
14 | xxx | yyy
15 | xxx | yyy
17 | xxx | yyy
19 | xxx | yyy
..............

What I want is to extract only the rows which have the value of the first column between 12 and 16, for example. I tried with numpy.where(), but I didn't really manage to.

So what I want is something that returns:

13 | xxx | yyy
14 | xxx | yyy
15 | xxx | yyy

rlaverde · Accepted Answer · 2014-04-16 22:44:07Z

2

You can try something like this

b = your_array[:,0]
condition = (b>12) & (b<16)

new_array = your_array[condition]

edited Apr 16, 2014 at 22:44

answered Apr 16, 2014 at 22:38

rlaverde

1567 bronze badges

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

Comments

joc · Accepted Answer · 2014-04-16 22:35:47Z

0

Using filter and lambda:

filtered_array = numpy.array(filter(lambda row: 12<row[0]<16, array))

answered Apr 16, 2014 at 22:35

joc

1,0887 silver badges8 bronze badges

Collectives™ on Stack Overflow

Numpy: get only the array rows that have a value in an interval

2 Answers 2

Comments

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

2 Answers 2

Comments

Comments

Your Answer

Sign up or log in

Post as a guest

Related