numpy: search of the first and last index in an array

Question

I use Python with numpy.

I have a numpy array b:

b = np.array([True,True,True,False,False,True,True,False,True,True,True,True,False])

I need to find the first and last index where b is equal True.

For this exsample:

out_index: [0,2]
           [5,6]
           [8,11]

Can someone please suggest, how do I get out_index?

HYRY · Accepted Answer · 2013-02-28 12:10:12Z

6

b = np.array([True,True,True,False,False,True,True,False,True,True,True,True,False])
idx = np.argwhere(np.diff(np.r_[False, b, False])).reshape(-1, 2)
idx[:, 1] -= 1
print idx

output:

[[ 0  2]
 [ 5  6]
 [ 8 11]]

answered Feb 28, 2013 at 12:10

HYRY

97.9k28 gold badges197 silver badges192 bronze badges

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

Collectives™ on Stack Overflow

numpy: search of the first and last index in an array

1 Answer 1

Comments

Your Answer

Linked

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

Comments

Your Answer

Sign up or log in

Post as a guest

Linked

Related