python sorted key

CV計算機視覺工程師發表於2020-11-13

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]]

相關文章