[python] 從numpy數組中取出具有給定步幅/步長的子陣列


0 Answers

Question

可以說我有一個Python Numpy數組數組a。

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

我想從這個長度為5的數組中創建一個子序列矩陣,結果矩陣因此如下所示:

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

一種可能的實現方法是使用for循環。

result_matrix = np.zeros((3, 5))
for i in range(0, len(a), 3):
  result_matrix[i] = a[i:i+5]

有沒有一個更清潔的方式來實現這個是Numpy?




Related