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