使用NumPy的argsort函数对数组按照每个向量的第1个元素进行排序

import numpy as np

# 创建包含向量的 NumPy 数组
array = np.array([[3, 2, 4], [1, 6, 5], [9, 7, 8], [0, 1, 2]])

# 使用 sort 函数按照每个向量的第一个元素排序
sorted_array = array[array[:, 0].argsort()]

print(sorted_array)

在这个代码示例中，我们直接将array数组传递给array[:, 0].argsort()作为sort函数的参数。array[:, 0]用于获取每个向量的第一个元素，然后argsort()函数用于返回按照这些元素升序排序的索引。

最后，我们使用这些排序后的索引将原始数组array重新排序，并将结果存储在sorted_array中。