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