这个特定列表理解的 Numpy 版本

Question

所以几天前我需要在这个线程中进行特定的列表理解：Selecting a subset of integers given two lists of end points

我得到了满意的答案。现在快进，我需要以某种方式提高性能，因为我正在处理的工作涉及循环它，并且在每次迭代中，这些端点数组的长度至少为几千。

所以我的问题是 numpy 包中是否有任何功能可以完成工作但速度更快？我查看了 numpy 的 linspace、repeat、arange 和其他东西，但找不到任何突破。如果有办法更快地完成工作，你们能给我指路吗？

提前谢谢你们。

Answer 1

如果您仍然对它感兴趣，您可以去掉一个 for 循环，并结合使用 numpy.arange() 与列表理解和 numpy.hstack() 来获得所需的内容。话虽如此，我们仍然需要至少一个 for 循环来完成这项工作（因为 range 和 arange 都不接受一系列端点）

t1 = [0,13,22]
t2 = [4,14,25]
np.hstack([np.arange(r[0], r[1]+1)  for r in zip(t1, t2)])

# outputs
array([ 0,  1,  2,  3,  4, 13, 14, 22, 23, 24, 25])

但是，我不知道这对于您的具体情况会有多高的性能。