numpy.swapaxes 函数

此函数交换数组的两个轴。 对于 1.10 之后的 NumPy 版本,返回一个交换数组的视图。 该函数接受以下参数。

numpy.swapaxes(arr, axis1, axis2)

这里,

序号 参数 & 描述
1

arr

要交换轴的输入数组

2

axis1

第一个坐标轴对应的int

3

axis2

第二坐标轴对应的int


示例

# It creates a 3 dimensional ndarray 
import numpy as np 
a = np.arange(8).reshape(2,2,2) 

print 'The original array:' 
print a 
print '\n'  
# now swap numbers between axis 0 (along depth) and axis 2 (along width) 

print 'The array after applying the swapaxes function:' 
print np.swapaxes(a, 2, 0)

它的输出将如下所示 −

The original array:
[[[0 1]
 [2 3]]

 [[4 5]
  [6 7]]]

The array after applying the swapaxes function:
[[[0 4]
 [2 6]]
 
 [[1 5]
  [3 7]]]

❮ NumPy 数组操作