在Numpy中移動非重疊窗口

Question

在numpy數組上移動窗口的最佳方式是什麼，以便每個單獨的塊不與前一個塊重疊，並且塊之間有1個元素間隙？我想我應該使用np.hstack，但我無法弄清楚切片模式。

基本上我所尋找的是這樣的：

a = np.array([0, 1, 2, 3, 4, 5, 6, 7, 8, 9]) 
result = np.array([[0, 1, 2, 3], 
        [5, 6, 7, 8]) 

謝謝！

Answer 1

您想要在簡短示例中實現的功能可以通過重新整形數組，然後移除最後一列來創建「間隙」來完成。

import numpy as np 

a = np.array([0, 1, 2, 3, 4, 5, 6, 7, 8, 9]) 

# get length of flat array 
a_length, = a.shape 

# reshape array 
#(column by row must respect number of elements) 
b = a.reshape((2, a_length/2)) 

# assign array except last column to a variable 
result = b[:,:-1] 
print result 

這是否足夠普遍以回答您的問題？