numpy基本操作--数组拆分

import numpy as np

A = np.arange(1, 13).reshape(6, 2)


B = np.vsplit(A, 3)

C = A.T

D = np.hsplit(C, 3)
print(A, B, C, D)

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