使用numpy划分矩阵中的行

import numpy as np

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

sub_matrices = np.split(matrix, 2)

for sub_matrix in sub_matrices:
    print(sub_matrix)

import numpy as np

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

sub_matrices = np.split(matrix, 2)

for sub_matrix in sub_matrices:
    print(sub_matrix)

[[1 2 3]
 [4 5 6]]
[[ 7  8  9]
 [10 11 12]]