按照某一列的大小重新排列numpy矩阵的所有行

偶然间有这么个需求，按照矩阵某一列把所有行重新排列，比方说原始矩阵为：

[[3, 9, 1], 
 [6, 2, 4],
 [9, 1, 7]]

要按照第二列从小到大的排列，在网上只搜索到只排序第二列的方法即：

[[3, 1, 1], 
 [6, 2, 4], 
 [9, 9, 7]]

其他列没有“随动”，而我想要实现的结果是：

[[9, 1, 7]，
 [6, 2, 4],
 [3, 9, 1]]

好了，上代码：

import numpy as np  
  
# 定义矩阵  
matrix = np.array([[3, 9, 1], [6, 2, 4], [9, 1, 7]])  
  
# 选择要排序的列（例如第二列）  
column_index = 1  
  
# 对该列的值进行排序，并获取排序后的索引  
sorted_indices = np.argsort(matrix[:, column_index])  
  
# 使用排序后的索引对原始矩阵进行重新排序  
sorted_matrix = matrix[sorted_indices]  
  
print(sorted_matrix)

OK