Как отсортировать массив 3D numpy по столбцу?

Question

Для двумерного массива, а, мы можем отсортировать, используя

a = a[a[:, 0].argsort]

, если мы хотим отсортировать по столбцу 0. Как сделать аналогичную вещь для трехмерной матрицы? Если у нас есть

a = [[[ 1.,  2., 10.],
         [ 4.,  5.,  6.],
         [ 2.,  3.,  4.]],

        [[ 2.,  3.,  4.],
         [ 4.,  5.,  6.],
         [ 1.,  2.,  3.]],

        [[ 1.,  2.,  3.],
         [ 4.,  5.,  6.],
         [ 2.,  3.,  4.]]]

И мы хотим отсортировать по первому столбцу каждого элемента 3d-матрицы?