numpy.swapaxes
numpy.swapaxes(a, axis1, axis2) [source]
互换数组的两个轴。
参数 ： | a ：array_like 输入数组。 axis1 ： 第一轴。 axis2 ： 第二轴。 |
返回值 ： | a_swapped ：ndarray 对于NumPy> = 1.10.0，如果a是一个 则返回a的视图；否则，返回a。 否则，将创建一个新数组。 对于较早的NumPy版本， 仅当更改轴顺序时才返回a的视图，否则返回输入数组。 |
例子
1）交换二维数组的轴
import numpy as np
a = np.array([[1, 2, 3],
[4, 5, 6]])
print("原数组：")
print(a)
# 交换轴0（行）和轴1（列）
b = np.swapaxes(a, 0, 1)
print("交换后的数组：")
print(b)
2）交换三维数组的轴
import numpy as np
a = np.array([[[1, 2], [3, 4]],
[[5, 6], [7, 8]]])
print("原数组：")
print(a)
# 交换轴0和轴2
b = np.swapaxes(a, 0, 2)
print("交换后的数组：")
print(b)
3）使用示例
import numpy as np
# 交换二维数组的轴
x = np.array([[1, 2, 3]])
print("原数组：")
print(x)
# 交换轴0和轴1
result = np.swapaxes(x, 0, 1)
print("交换后的数组：")
print(result)
# 交换三维数组的轴
x = np.array([[[0, 1], [2, 3]],
[[4, 5], [6, 7]]])
print("原数组：")
print(x)
# 交换轴0和轴2
result = np.swapaxes(x, 0, 2)
print("交换后的数组：")
print(result)