[Numpy-discussion] ‎ Find contiguous sequences of booleans in arrays

David Cournapeau david at ar.media.kyoto-u.ac.jp
Mon Sep 17 06:33:03 EDT 2007


Kurdt Bane wrote:
> Hi to all,
>
> I've got an 1-D array of bools and I'd like to find the length of the 
> first contiguous sequence of True values, starting from position [0] 
> of the array.
> (That's equivalent to find the position of the first occurrence of 
> False in the array).
One possibility would be to find the first True and then use cumproduct 
to get the length, and then create a view with the remaining elements, 
recursively. But this may be slow. You may be a bit smarter to get the 
index of the first True value for each sub-sequence by multiplying the 
array by itself, shifted by one, and using xor. Whether this is pythonic 
and clean, I don't know :)

cheers,

David



More information about the NumPy-Discussion mailing list