Numpy多维数组的轴对换 transpose

转置中,transpose方法返回的是源数据的视图,就是说修改了视图就会把源数据也改了。


高维数组,transpose的方法如下展示


创建一个形状为(2,2,4)的三维数组:

>>> arr = np.arange(16).reshape((2,2,4))

>>> arr

array([[[ 0,  1,  2,  3],
        [ 4,  5,  6,  7]],

       [[ 8,  9, 10, 11],
        [12, 13, 14, 15]]])

transpose需要放入一个由轴编号组成的元组才能变换,意思就是放入需要改变的shape

我需要把 0轴 和 1轴 对调,2轴不变,就是放入(1,0,2)

>>> arr.transpose((1,0,2))

array([[[ 0,  1,  2,  3],
        [ 8,  9, 10, 11]],

       [[ 4,  5,  6,  7],
        [12, 13, 14, 15]]])

这里就需要解释一下了,例如第二行的8这个数,这个位置原来是4,4对应的坐标为(0,1,0)——第3维的第一个元素中的第二个元素的第一个。经过transpose转换之后,这个位置对应的便是(1,0,0)中的8,如此类推


















你可能感兴趣的:(Numpy多维数组的轴对换 transpose)