如何像這樣用 numpy 重塑數組：

Question

我有這個：

array([[0, 0, 1, 1, 2, 2, 3, 3],
       [0, 0, 1, 1, 2, 2, 3, 3]])

我想像這樣重塑我的數組：

array([[0, 0, 1, 1],
       [0, 0, 1, 1],
       [2, 2, 3, 3],
       [2, 2, 3, 3]])

我如何使用 python numpy 做到這一點？

Answer 1

您可以拆分和連接：

a = np.array([[0, 0, 1, 1, 2, 2, 3, 3],
              [0, 0, 1, 1, 2, 2, 3, 3]])

cols = a.shape[1] // 2
np.concatenate((a[:,:cols], a[:,cols:]))

#[[0 0 1 1]
# [0 0 1 1]
# [2 2 3 3]
# [2 2 3 3]]

Answer 2

您可以在重塑后簡單地交換行。

a= np.array([[0, 0, 1, 1, 2, 2, 3, 3],
       [0, 0, 1, 1, 2, 2, 3, 3]]).reshape(4,4)

a[[1,2]] = a[[2,1]]

Output：

array([[0, 0, 1, 1],
       [0, 0, 1, 1],
       [2, 2, 3, 3],
       [2, 2, 3, 3]])