Eu tenho uma matriz numpy 3D que preciso remodelar e organizar. Por exemplo, eu tenhox=np.array([np.array([np.array([1,0,1]),np.array([1,1,1]),np.array([0,1,0]),np.array([1,1,0])]),np.array([np.array([0,0,1]),np.array([0,0,0]),np.array([0,1,1]),np.array([1,0,0])]),np.array([np.array([1,0,0]),np.array([1,0,1]),np.array([1,1,1]),np.array([0,0,0])])])

Que é uma forma de (3,4,3), ao imprimi-lo, recebo:

array([[[1, 0, 1],
        [1, 1, 1],
        [0, 1, 0],
        [1, 1, 0]],

       [[0, 0, 1],
        [0, 0, 0],
        [0, 1, 1],
        [1, 0, 0]],

       [[1, 0, 0],
        [1, 0, 1],
        [1, 1, 1],
        [0, 0, 0]]])

Agora preciso remodelar essa matriz para um(4,3,3) selecionando o mesmo índice em cada sub-matriz e reunindo-os para terminar com algo assim:

array([[[1,0,1],[0,0,1],[1,0,0]],
[[1,1,1],[0,0,0],[1,0,1]],
[[0,1,0],[0,1,1],[1,1,1]],
[[1,1,0],[1,0,0],[0,0,0]]]

eu tenteireshape, todos os tipos de empilhamento e nada funcionou (organizou a matriz como eu preciso). Eu sei que posso fazê-lo manualmente, mas para matrizes grandes manualmente não é uma opção.

Qualquer ajuda será muito apreciada. obrigado