NumPy 函数

NumPy swapaxes() 函数用于交换数组的两个轴。

语法

numpy.swapaxes(a, axis1, axis2)

参数

a必填。 指定输入数组。
axis1必填。 指定第一个轴。
axis2必填。 指定第二个轴。

返回值

如果a是一个ndarray，然后返回a的视图；否则会创建一个新数组。

示例:

在下面的示例中，swapaxes() 函数用于交换数组的两个轴。

import numpy as np

x = np.array([[1,2,3]])
y = np.array([[[1,2,3]],
              [[4,5,6]]])

#打印x
print("x =")
print(x)

#交换x轴
print("\nnp.swapaxes(x,0,1) =")
print(np.swapaxes(x,0,1))

#print y
print("\ny =")
print(y)

#交换 y 轴
print("\nnp.swapaxes(y,0,2) =")
print(np.swapaxes(y,0,2))

上述代码的输出将是:

x =
[[1 2 3]]

np.swapaxes(x,0,1) =
[[1]
 [2]
 [3]]

y =
[[[1 2 3]]

 [[4 5 6]]]

np.swapaxes(y,0,2) =
[[[1 4]]

 [[2 5]]

 [[3 6]]]