python sort搭配lambda實現多欄位排序
阿新 • • 發佈:2018-12-15
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')]
快去試一試吧!!!