Python 如何重新排列3d numpy阵列？

我有一个numpy ndarray，类似这样的东西（我的数组要大得多，我给出这个只是为了解释我需要做什么）：

我正在寻找一种优雅的方法来重新排列阵列，以获得如下效果：

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)

---

很抱歉我的错误：）你真的需要重新排列数组吗，或者是一个

numpy

lazy转置（基本上只是重命名轴）可以吗？

a.transpose(1,2,0)
# or np.moveaxis(a,0,2)