Переставить матрицу Numpy

Question

У меня есть матрица, как показано ниже с формой (2, 2, 1).

[[[1]
  [3]]

 [[2]
  [4]]]

Есть ли простой способ / функция для перестановки элементов таким образом, что она становится:

[[[1]
  [2]]

 [[3]
  [4]]]

Спасибо!

Answer 1

Вы можете использовать swapaxes для чередования двух первых осей массива:

a.swapaxes(0,1)

array([[[1],
        [2]],

       [[3],
        [4]]])