A numpy.arange style function with array inputs

Question

I have numpy arrays of start indices and end indices from which I'd like to construct a flattened array of ranges.

e.g. with inputs

s = np.array([1,2,3])
e = np.array([4,5,10])

and output

array([1,2,3,2,3,4,3,4,5,6,7,8,9])

any way to do this efficiently?

tkerwin · Accepted Answer · 2011-05-17 20:49:15Z

1

How about just

np.concatenate([np.arange(x, y) for x, y in zip(s, e)])

answered May 17, 2011 at 20:49

tkerwin

9,7891 gold badge34 silver badges47 bronze badges

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

1 Comment

Joe Kington Over a year ago

@Chris - While @tkerwin's answer is perfectly suited to your case, you might also want to have a look at np.r_. It's a convenient way to manually build up a sequence like this. np.r_[1:4, 2:5, 3:10] will create the same sequence. It's probably not what you want to use in this case, but is very convenient in some cases, and it's good to know about.

Collectives™ on Stack Overflow

A numpy.arange style function with array inputs

1 Answer 1

1 Comment

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

1 Comment

Your Answer

Sign up or log in

Post as a guest

Related