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]]]