我有一个numpy ndarray，像这样（我的数组更大，我只是为了解释我需要做的事情而已）：
a = [[[ 0 1 2]
[ 3 4 5]
[ 6 7 8]]
[[ 9 10 11]
[12 13 14]
[15 16 17]]
[[18 19 20]
[21 22 23]
[24 25 26]]]
我正在寻找一种优雅的方式来重新排列数组，以得到这样的东西：
a_new=[[[ 0 9 18]
[ 1 10 19]
[ 2 11 20]]
[[ 3 12 21]
[ 4 13 22]
[ 5 14 23]]
[[ 6 15 24]
[ 7 16 25]
[ 8 17 26]]]
只需置换轴-
a.transpose(1,2,0)
# or np.moveaxis(a,0,2)
本文收集自互联网，转载请注明来源。
如有侵权，请联系[email protected] 删除。
我来说两句