Numpy：维度的转换

介绍

numpy.transpose 是一个非常有用的函数，用于改变数组的轴顺序。以下是 numpy.transpose 的详细用法介绍：

语法：

numpy.transpose(a, axes=None)

参数：

• a：输入的数组。
• axes：一个可选参数，用于指定新的轴的顺序。如果省略，则默认为 None，表示自动交换轴的顺序。

返回值：

返回一个新的数组，其轴的顺序由 axes 参数指定。

示例

1. 二维数组（矩阵）：

import numpy as np  
  
a = np.array([[1, 2], [3, 4], [5, 6]])  
print("Original:")  
print(a)  
print("Transpose:")  
print(np.transpose(a))

输出：

Original:  
[[1 2]  
 [3 4]  
 [5 6]]  
Transpose:  
[[1 3 5]  
 [2 4 6]]

2. 三维的维度变换

import numpy as np

a = np.arange(24).reshape(3, 4, 2)
print("Original:"+str(a.shape))

b = np.transpose(a, (1, 0, 2)) # 交换第一个和第二个轴，第三个轴保持不变
print("Transpose:"+str(b.shape))