NumPy如何交换数组轴

NumPy提供了np.swapaxes()函数来交换数组的轴。该函数接受两个参数，分别表示要交换的两个轴的索引。例如，对于一个二维数组，可以使用np.swapaxes(arr, 0, 1)来交换其行和列。

示例代码：

import numpy as np

arr = np.array([[1, 2, 3], [4, 5, 6]])
print("原始数组：")
print(arr)

arr_swapped = np.swapaxes(arr, 0, 1)
print("交换轴后的数组：")
print(arr_swapped)

输出结果：

原始数组：
[[1 2 3]
 [4 5 6]]
交换轴后的数组：
[[1 4]
 [2 5]
 [3 6]]