使用numpy获取按另一列排序的给定列值的行索引

import numpy as np

data = np.array([[1, 4, 7],
                 [2, 5, 8],
                 [3, 6, 9]])

column_to_sort = 1  # 指定要排序的列索引
sorted_indexes = np.argsort(data[:, column_to_sort])

print(sorted_indexes)

import numpy as np

data = np.array([[1, 4, 7],
                 [2, 5, 8],
                 [3, 6, 9]])

column_to_sort = 1  # 指定要排序的列索引
sorted_indexes = np.argsort(data[:, column_to_sort])

print(sorted_indexes)