带数组输入的numpy.arange风格函数

Question

我有一些numpy数组，里面存的是开始索引和结束索引，我想用这些索引来构建一个扁平化的范围数组。

比如，给定这些输入

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

想要得到这样的输出

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

有没有什么高效的方法可以做到这一点呢？

Answer 1

那这样怎么样呢

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