Boolean Indexing n-D Numpy Array

Question

I could not apply bool values into given data:

import numpy as np

bool_values = np.array([False, False, True, False, True])
data = np.array([[11.0, 22.0, 0.0, 44.0, 55.0],
                 [10.0, 20.0, 30.0, 40.0, 50.0]])

expected_answer = [[0,55], [30,50]]

I tried it as:

expected_answer = data[bool_values]

You mean [0.0, 55] right?

Ashwini Chaudhary
– Ashwini Chaudhary

2015-06-22 14:32:55 +00:00
Commented Jun 22, 2015 at 14:32 — Ashwini Chaudhary
– Ashwini Chaudhary, Commented Jun 22, 2015 at 14:32
Aha yes, now corrected

Roman
– Roman

2015-06-22 14:33:29 +00:00
Commented Jun 22, 2015 at 14:33 — Roman
– Roman, Commented Jun 22, 2015 at 14:33

EdChum · Accepted Answer · 2015-06-22 14:33:35Z

1

I think you wanted to do this:

In [28]:
bool_values = np.array([False, False, True, False, True])
data = np.array([[11.0, 22.0, 0.0, 44.0, 55.0],
 [10.0, 20.0, 30.0, 40.0, 50.0]])
data[:,bool_values]

Out[28]:
array([[  0.,  55.],
       [ 30.,  50.]])

The above applies the mask to every row

answered Jun 22, 2015 at 14:33

EdChum

397k204 gold badges836 silver badges583 bronze badges

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

Collectives™ on Stack Overflow

Boolean Indexing n-D Numpy Array

1 Answer 1

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

Comments

Your Answer

Sign up or log in

Post as a guest

Related