NumPy 数组操作 numpy.swapaxes
此函数交换数组的两个轴。对于1.10之后的NumPy版本,将返回交换后数组的视图。该函数接受以下参数。
numpy.swapaxes(arr, axis1, axis2)
在这里,
序号 | 参数及描述 |
---|---|
1 | arr 需要交换轴的输入数组 |
2 | axis1 对应第一个轴的整数 |
3 | axis2 对应第二个轴的整数 |
示例
# 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]]]