[Numpy-discussion] List/location of consecutive integers

Christopher Barker Chris.Barker@noaa....
Fri May 22 17:13:00 CDT 2009


David Warde-Farley wrote:
> I don't think this is very general:
> 
> In [53]: indices
> Out[53]:
> array([   -3,     1,     2,     3,     4,     5,     6,     7,     8,
>             9,   255,   256,   257,   258, 10001, 10002, 10003, 10004])
> 
> In [54]: breaks = diff(indices) != 1
> 
> In [55]: zip(indices[np.r_[True, breaks[:-1]]], indices[breaks])
> Out[55]: [(-3, -3), (1, 9), (255, 258)]

that's why I put a sys.maxint at the end of the series...


In [13]: indices = np.array([   -3,     1,     2,     3,     4,     5, 
    6,     7,     8,
             9,   255,   256,   257,   258, 10001, 10002, 10003, 10004, 
sys.maxint])

In [15]: breaks = np.diff(indices) != 1

In [16]: zip(indices[np.r_[True, breaks[:-1]]], indices[breaks])
Out[16]: [(-3, -3), (1, 9), (255, 258), (10001, 10004)]


Though that's probably not very robust!

-Chris



-- 
Christopher Barker, Ph.D.
Oceanographer

Emergency Response Division
NOAA/NOS/OR&R            (206) 526-6959   voice
7600 Sand Point Way NE   (206) 526-6329   fax
Seattle, WA  98115       (206) 526-6317   main reception

Chris.Barker@noaa.gov


More information about the Numpy-discussion mailing list