有没有其他方法可以按'column‘值对3D numpy数组进行排序？

import numpy as np

# 创建一个3D numpy数组
arr = np.array([
    [[4, 2, 1], [5, 3, 2], [2, 1, 4]],
    [[1, 3, 5], [3, 2, 4], [5, 4, 3]],
    [[3, 5, 2], [4, 1, 3], [1, 2, 5]]
])

# 按第一列的值对数组进行排序
sorted_indices = np.argsort(arr[:, :, 0], axis=0)
sorted_arr = arr[sorted_indices]

print(sorted_arr)

[[[1 3 5]
  [4 1 3]
  [5 4 3]]

 [[3 5 2]
  [5 3 2]
  [2 1 4]]

 [[4 2 1]
  [1 2 5]
  [3 2 4]]]