Convert numpy array from values to indices

Question

I have a numpy array a: [True, False, True, False, False, ...] And I like to have a numpy array that has the indices of the True and False values, i.e. [0, 2, ...] and [1, 3, 4, ...]

Sven Marnach · Accepted Answer · 2012-02-15 16:13:40Z

9

To get the indices of the True values in a, you can use

a.nonzero()

For the indices of the False values, use

(~a).nonzero()

answered Feb 15, 2012 at 16:13

Sven Marnach

608k123 gold badges966 silver badges865 bronze badges

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

Collectives™ on Stack Overflow

Convert numpy array from values to indices

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