多维数组翻转前几个维度
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]]])