python numpy 按照维度排序

你可以使用 numpy 的 sort 函数进行按维度排序。例如，如果你想按照第一维排序一个数组 arr，可以使用以下代码：

import numpy as np

arr = np.array([[3, 2, 1], [6, 5, 4], [9, 8, 7]])
sorted_arr = np.sort(arr, axis=0)
print(sorted_arr)

输出：

[[3 2 1]
 [6 5 4]
 [9 8 7]]

如果你想按照第二维排序，可以将 axis 参数设置为 1：

import numpy as np

arr = np.array([[3, 2, 1], [6, 5, 4], [9, 8, 7]])
sorted_arr = np.sort(arr, axis=1)
print(sorted_arr)

输出：

[[1 2 3]
 [4 5 6]
 [7 8 9]]

你还可以使用 argsort 函数获取排序后的索引，例如：

import numpy as np

arr = np.array([[3, 2, 1], [6, 5, 4], [9, 8, 7]])
sorted_indices = np.argsort(arr, axis=1)
print(sorted_indices)

输出：

[[2 1 0]
 [2 1 0]
 [2 1 0]]

这个结果表示第一行的元素按照从小到大的顺序排列后，索引为 2 的元素最小，索引为 0 的元素最大。