Python 二维数组排序:获取某列前三最大元素
使用 sorted() 函数对二维数组中的某一列进行排序,然后使用切片操作获取排序后的前3个元素。
下面是一个例子:
# 二维数组
array = [[1, 4, 7],
[2, 5, 8],
[3, 6, 9]]
# 按照第二列进行排序
sorted_array = sorted(array, key=lambda x: x[1])
# 输出排序后的前3个元素
top_3 = sorted_array[:3]
print(top_3)
输出:
[[1, 4, 7], [2, 5, 8], [3, 6, 9]]
在这个例子中,我们对二维数组 array 按照第二列进行排序,并将排序后的前3个元素存储在 top_3 变量中。最后打印出 top_3 的值。
原文地址: https://www.cveoy.top/t/topic/eHtC 著作权归作者所有。请勿转载和采集!