key 指定一个参数的函数,用于从每个列表元素中提取比较键
示例:创建由元组构成的列表:a = [('b',3), ('a',2), ('d',4), ('c',1)]
- 按照第一个元素排序
sorted(a, key=lambda x:x[0])
>>> [('a',2),('b',3),('c',1),('d',4)]
- 按照第二个元素排序
sorted(a, key=lambda x:x[1])
>>> [('c',1),('a',2),('b',3),('d',4)]
列题:
def get_distance(num): return abs(num[0])**2 +abs(num[1])**2 s = [[4,5], [0,1],[2,3],[3,4]] s = sorted(s,key=lambda x: get_distance(x)) print(s)
output:
[[0, 1], [2, 3], [3, 4], [4, 5]]