arrays - 如何在numpy中创建连续数字数组？

Question

我希望生成一个数组，如下所示:

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

这是我在 NumPy 中生成的方法:

import numpy as np
a = np.arange(1, 10)
b = np.arange(2, 11)
np.stack((a, b), axis=1)

NumPy 中有直接执行此操作的函数吗？

Answer 1

我能想到的最短答案，使用 broadcasting黑魔法:

# Solution 1:
np.r_[:9][:,None]+[1,2]
# Solution 2:
np.r_['c',:9]+[1,2]

或者也使用np.r_但这次没有广播:

# Solution 3:
np.r_['1,2,0', :10, 1:11]

每个解决方案都会产生预期的结果:

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