如何按列对nx3 numpy数组进行排序，但它会记住该行中的数据？

import numpy as np

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

sorted_indices = np.argsort(array[:, 0])  # 按第一列排序，返回排序后的索引
sorted_array = array[sorted_indices]  # 根据索引重新排序数组

sorted_array[:, 0] = array[:, 0]  # 保留第一列原始数据

import numpy as np

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

sorted_indices = np.argsort(array[:, 0])
sorted_array = array[sorted_indices]

sorted_array[:, 0] = array[:, 0]

print(sorted_array)