多维数组翻转前几个维度

NumPy publisher01 56℃

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

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

转载请注明:Python量化投资 » 多维数组翻转前几个维度

喜欢 (0)or分享 (0)