numpy 数组切片，没有 for 循环

Question

我有一个问题，如何从 numpy 数组生成切片数组。我可以用for循环来做，我想做的是：

def calc_matrix(arr, k): 
    x = np.zeros((len(arr) - k + 1, k))
    for i in range(len(arr) - k + 1):
        x[i, :] = arr[i: (i+k)]
    return x

Answer 1

您可以从arr 的 Hankel 矩阵（以及任意第二个参数；下面我们将其完全省略，在这种情况下默认使用零）中删除您想要的结果：

>>> import numpy as np                                                                                              
>>> from scipy import linalg
>>>
>>> arr = np.arange(10)**2 % 7   ##  just a random example      
>>> arr
array([0, 1, 4, 2, 2, 4, 1, 0, 1, 4])                                            
>>> k = 4                                                                                                           
>>> linalg.hankel(arr)[:arr.size-k+1, :k]                                              
array([[0, 1, 4, 2],                                                                                                
       [1, 4, 2, 2],                                                                                                
       [4, 2, 2, 4],                                                                                                
       [2, 2, 4, 1],                                                                                                
       [2, 4, 1, 0],
       [4, 1, 0, 1],
       [1, 0, 1, 4]])

或者你可以使用http://scikit-image.org/docs/dev/api/skimage.util.html#skimage.util.view_as_windows。