NumPy numpy.swapaxes 函数
-
简述
此函数交换数组的两个轴。对于 1.10 之后的 NumPy 版本,将返回交换数组的视图。该函数采用以下参数。numpy.swapaxes(arr, axis1, axis2)
参数说明序号 参数及说明 1 arr要交换其轴的输入数组2 axis1对应于第一个轴的 int3 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]]]