numpy 二维、高维矩阵重新排列轴与数据（轴转换）

使用的函数是numpy.transpose，例子如下：

import numpy as np

array = np.arange(24).reshape((1, 2, 3, 4))
print('array : ', array.shape)

返回的array为(1, 2, 3, 4)

array2 = array.transpose((0, 1, 2, 3))
print('array2 : ', array2.shape)

维度变成了(1, 2, 3, 4)

array3 = array.transpose((2, 3, 0, 1))
print('array3 : ', array3.shape)

维度变成了(3, 4, 1, 2)

参考文档

numpy.transpose 官方DOC：https://numpy.org/doc/stable/reference/generated/numpy.transpose.html