Python:使用lambda對列表(list)和字典(dict)排序

轉自Penguin博客

自己驗證實現並整理，對列表和字典排序時lambda的用法。

函數介紹

sorted()

sorted函數：

sorted(iterable[,key][,reverse])

函數接收三個參數：排序的變量、排序的規則、升降序選擇

lambda

lambda 匿名函數
一般形式

lambda arguments: expression

寫成函數形式

def <lambda>(arguments):
        return expression

排序

1 簡單列表(list)排序

list = ['a', 'b', 'c']
print(sorted(list))
# ['a', 'b', 'c']
print(sorted(list, reverse=True))
# ['c', 'b', 'a']

2 字典(dict)的鍵(key)排序

dict = {'c': 1, 'b': 2, 'a': 3}
print(sorted(dict))
# ['a', 'b', 'c']
print(sorted(dict, reverse=True))
# ['c', 'b', 'a']

3 字典(dict)的值(value)排序

使用lambda函數讓key為字典值即dict[]

dict = {'c': 1, 'b': 2, 'a': 3}
print(sorted(dict, key=lambda k: dict[k]))
# ['c', 'b', 'a']
print(sorted(dict, key=lambda k: dict[k], reverse=True))
# ['a', 'b', 'c']

4列表(list)內嵌套列表(list)排序

list = [[4, 2, 9], [1, 5, 6], [7, 8, 3]]
# 以列表中列表的第一個數排序
print(sorted(list, key=lambda k: k[0]))
# [[1, 5, 6], [4, 2, 9], [7, 8, 3]]
# 以列表中列表的第二個數排序
print(sorted(list, key=lambda k: k[1]))
# [[4, 2, 9], [1, 5, 6], [7, 8, 3]]
# 以列表中列表的第三個數排序
print(sorted(list, key=lambda k: k[2]))
# [[7, 8, 3], [1, 5, 6], [4, 2, 9]]
# 以列表中列表的第一個數排序，且降序
print(sorted(list, key=lambda k: k[0], reverse=True))
# [[7, 8, 3], [4, 2, 9], [1, 5, 6]]

5字典(dict)內嵌套字典(dict)排序

dict = {
        'a': {'x': 3, 'y': 2, 'z': 1},
        'b': {'x': 2, 'y': 1, 'z': 3},
        'c': {'x': 1, 'y': 3, 'z': 2}
}
# 以內部字典的'x'對應的值排序
print(sorted(dict, key=lambda k: dict[k]['x']))
# ['c', 'b', 'a']
# 以內部字典的'y'對應的值排序
print(sorted(dict, key=lambda k: dict[k]['y']))
# ['b', 'a', 'c']
# 以內部字典的'z'對應的值排序
print(sorted(dict, key=lambda k: dict[k]['z']))
# ['a', 'c', 'b']
# 以內部字典的'x'對應的值排序,並降序
print(sorted(dict, key=lambda k: dict[k]['x'], reverse=True))
# ['a', 'b', 'c']

6 列表(list)中嵌套字典(dict)排序

list = [
    {'x': 3, 'y': 2, 'z': 1},
    {'x': 2, 'y': 1, 'z': 3},
    {'x': 1, 'y': 3, 'z': 2},
]
print(sorted(list, key=lambda k: k['x']))
# [{'z': 2, 'x': 1, 'y': 3}, {'z': 3, 'x': 2, 'y': 1}, {'z': 1, 'x': 3, 'y': 2}]
print(sorted(list, key=lambda k: k['y']))
# [{'z': 3, 'x': 2, 'y': 1}, {'z': 1, 'x': 3, 'y': 2}, {'z': 2, 'x': 1, 'y': 3}]
print(sorted(list, key=lambda k: k['z']))
# [{'z': 1, 'x': 3, 'y': 2}, {'z': 2, 'x': 1, 'y': 3}, {'z': 3, 'x': 2, 'y': 1}]
print(sorted(list, key=lambda k: k['x'], reverse=True))
# [{'z': 1, 'x': 3, 'y': 2}, {'z': 3, 'x': 2, 'y': 1}, {'z': 2, 'x': 1, 'y': 3}]

7字典(dict)中嵌套列表(list)排序

dic = {
    'a': [1, 2, 3],
    'b': [2, 1, 3],
    'c': [3, 1, 2],
}
print(sorted(dic, key=lambda k: dic[k][0]))
# ['a', 'b', 'c']
print(sorted(dic, key=lambda k: dic[k][1]))
# ['b', 'c', 'a']
print(sorted(dic, key=lambda k: dic[k][2]))
# ['c', 'b', 'a']
print(sorted(dic, key=lambda k: dic[k][0], reverse=True))
# ['c', 'b', 'a']

方法解釋

依據lambda的解釋，關注的主要是上述例子中k: 之后的表達式，其中k應該代表sorted()函數默認的key值。
比如：在3中，字典默認排序key是字典的鍵，所以lambda中k代表字典的鍵，想以字典的值排序，就應該是讓排序中的key=dict[k]
在4中，列表默認排序key是列表中的元素，此處列表中的元素依舊是嵌入的列表，所以排序key取嵌入列表的第一項時，可以key=k[0]