我有一个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]]]
答案 0 :(得分:3)
仅置换轴-
a.transpose(1,2,0)
# or np.moveaxis(a,0,2)