Python numpy.swapaxes()函数
numpy.swapaxes()函数互换一个数组的两个轴。
语法: numpy.swapaxes(arr, axis1, axis2)
参数 :
arr : [array_like] 输入阵列。
axis1 : [int] 第一个轴。
axis2 : [int] 第二轴。
返回 : [ndarray] 在早期的NumPy版本中,只有当轴的顺序被改变时才会返回arr的视图,否则会返回输入数组。对于NumPy >= 1.10.0,如果arr是一个ndarray,那么将返回arr的视图;否则将创建一个新数组。
代码#1:
# Python program explaining
# numpy.swapaxes() function
# importing numpy as geek
import numpy as geek
arr = geek.array([[2, 4, 6]])
gfg = geek.swapaxes(arr, 0, 1)
print (gfg)
输出 :
[[2]
[4]
[6]]
代码#2:
# Python program explaining
# numpy.swapaxes() function
# importing numpy as geek
import numpy as geek
arr = geek.array([[[0, 1], [2, 3]], [[4, 5], [6, 7]]])
gfg = geek.swapaxes(arr, 0, 2)
print (gfg)
输出 :
[[[0 4]
[2 6]]
[[1 5]
[3 7]]]