python碎碎念

多维数组翻转前几个维度

2019-06-12  本文已影响0人  你说你要一场

比如处理照片时想翻转前两维以便于更好处理。

numpy有两种处理方式。

>>> import  numpy as np
>>> aa=np.arange((12))
>>> aa
array([ 0,  1,  2,  3,  4,  5,  6,  7,  8,  9, 10, 11])
>>> aa=aa.rshape((2,3,2))
>>> aa
array([[[ 0,  1],
        [ 2,  3],
        [ 4,  5]],

       [[ 6,  7],
        [ 8,  9],
        [10, 11]]])
>>> cc=aa.swapaxes(0,1)
>>> cc
array([[[ 0,  1],
        [ 6,  7]],

       [[ 2,  3],
        [ 8,  9]],

       [[ 4,  5],
        [10, 11]]])
>>> np.transpose(aa,[1,0,2])
array([[[ 0,  1],
        [ 6,  7]],

       [[ 2,  3],
        [ 8,  9]],

       [[ 4,  5],
        [10, 11]]])

上一篇下一篇

猜你喜欢

热点阅读