假设我们要对一个二维数组arr按照第一列进行升序排序,可以使用Python内置的sorted函数以及lambda表达式实现:
arr = [[3, 2, 1], [1, 4, 6], [2, 3, 9], [5, 1, 3]]
sorted_arr = sorted(arr, key=lambda x: x[0])
print(sorted_arr)
输出结果为:
[[1, 4, 6], [2, 3, 9], [3, 2, 1], [5, 1, 3]]
其中key=lambda x: x[0]表示按照每个元素的第一个值进行排序。
如果要按照第一列进行降序排序,只需要在lambda表达式中加一个负号即可:
sorted_arr = sorted(arr, key=lambda x: -x[0])
print(sorted_arr)
输出结果为:
[[5, 1, 3], [3, 2, 1], [2, 3, 9], [1, 4, 6]]
上一篇:按第一列聚类