在数组中转置嵌套数组的最佳方法比 for 循环

Best way to transpose nested arrays within array than a for loop

我有一个形状为 (43200, 28, 28) 的 numpy 数组,其中包含 43200 张大小为 28x28 的图像。 我想转置这些图像中的每一个,但找不到比

这样的循环更好的方法
for i in range(arr.shape[0]):
  arr[i] = arr[i].T

有没有更高效的方法?

你可以保留0轴,但调换后两个

arr.transpose(0, 2, 1)