Python 如何对角镜像NX3 numpy阵列

除了最低级别的条目是三个数组而不是标量值之外，如何以与二维数组类似的方式转换三维数组

这就是我的意思：

M = [[[0,0,0][1,1,1][2,2,2]]
     [[0,0,0][0,0,0][3,3,3]]
     [[0,0,0][0,0,0][0,0,0]]]

N = some_operation(M)

N = [[[0,0,0][0,0,0][0,0,0]]
     [[1,1,1][0,0,0][0,0,0]]
     [[2,2,2][3,3,3][0,0,0]]]

我有一个python代码示例，它也展示了我的意思：

将numpy导入为np
M=np.数组（[[0,0,0]，[1,1,1]，[2,2,2]，[[0,0,0]，[0,0,0]，[3,3,3]，[[0,0,0]，[0,0,0]，[0,0,0]]））
N=np.数组（[[0,0,0]，[0,0,0]，[0,0,0]]，[[1,1,1]，[0,0,0]，[0,0,0]]，[2,2,2]，[3,3,3]，[0,0]]
打印（M）
打印（'\n\n'）
打印（M_翻转）

np.transpose（）

函数似乎不适合我的情况

提前感谢。

只需使用-

或与-

使用

np.rollaxis

-

N = np.rollaxis(M,1,0)

N = np.moveaxis(M,0,1)

N = np.rollaxis(M,1,0)