从一维数组构建高效的Numpy 2D数组

Question

我有一个像这样的数组：A = array([1,2,3,4,5,6,7,8,9,10])我试图得到这样的数组：B = array([[1,2,3],          [2,3,4],          [3,4,5],          [4,5,6]])每行（具有固定的任意宽度）每行偏移一个。A的数组是10k记录长，我试图在Numpy中找到一种有效的方法。目前，我正在使用vstack和for循环，这很慢。有没有更快的方法？编辑：width = 3 # fixed arbitrary widthlength = 10000 # length of A which I wish to useB = A[0:length + 1]for i in range (1, length):    B = np.vstack((B, A[i, i + width + 1]))