Python 使用 reshape 函数重排三维数组

import numpy as np

# 创建三维数组A
A = np.array([[[1, 2], [3, 4]], [[5, 6], [7, 8]], [[9, 10], [11, 12]]])

# 重排为数组B
B = np.reshape(A, (3, 2, 2))

print(B)

输出：

array([[[ 1,  2],
        [ 3,  4]],

       [[ 5,  6],
        [ 7,  8]],

       [[ 9, 10],
        [11, 12]]])