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]]