在numpy数组中转置子矩阵的有效方法

Question

我有一个非常大的numpy矩阵数组，其结构如下：

np.array([
           [[1, 2],
            [3, 4]],
           [[5, 6],
            [7, 8]],
         ])

我的预期输出是

np.array([
           [[1, 3],
            [2, 4]],
           [[5, 7],
            [6, 8]],
         ])

我知道使用列表理解或for循环的方法，然后再次创建numpy数组，但涉及从列表创建numpy的方法对我的数据来说太慢了。而vectorize似乎只对数字起作用。

当我使用普通的python列表或其他语言时，我可以将转置函数映射到列表中，但似乎numpy中没有类似的函数。

我怎样才能有效地做到这一点？