Python function to find indexes of 1s in binary array and

Question

I have an array which looks like this

[1, 0, 1 , 0 , 0, 1]

And I want to get those indexes that have 1 in it. So here I would get an array of <

Answer 1

There is np.where, np.argwhere, np.nonzero and np.flatnonzero. Take your pick.

np.where and np.nonzero are as far as I know equivalent, only there is also a three argument version of where that does something different. np.argwhere is the transpose and np.flatnonzero gives flat indices.