1. 程式人生 > >python sort搭配lambda實現多欄位排序

python sort搭配lambda實現多欄位排序

sort的常用方法你一定都會了,但是如果要排序的物件的每個元素又含有多個元素,要按指定的內部某個元素排序該怎麼實現呢?正常是這樣來做的:

>>> def return_item(item):
...     return item[1]
>>> pairs = [(1, 'one'), (2, 'two'), (3, 'three'), (4, 'four')]
>>> pairs.sort(key=return_item)

如果這樣做,那就要多定義一個函式,是不是感覺很low,那麼就使用lambda吧!

lambda是用來建立簡單邏輯函式的一個優雅的方法,例如:

>>> def make_incrementor(n):
...     return lambda x: x + n
...
>>> f = make_incrementor(42)
>>> f(0)
42
>>> f(1)
43

搭配sort的效果是這樣的:

>>> pairs = [(1, 'one'), (2, 'two'), (3, 'three'), (4, 'four')]
>>> pairs.sort(key=lambda pair: pair[1])
>>> pairs
[(4, 'four'), (1, 'one'), (3, 'three'), (2, 'two')]

如果按兩個元素排序,是這樣的:

>>> pairs = [(1, 'one'), (2, 'two'), (3, 'three'), (4, 'four')]
>>> pairs.sort(key=lambda pair: (pair[0], pair[1]))
>>> pairs
[(1, 'one'), (2, 'two'), (3, 'three'), (4, 'four')]

快去試一試吧!!!