寫一個刪除列表中重復(fù)元素的函數(shù),要求去重后元素相對位置保持不變。
點評:這個題目在初中級Python崗位面試的時候經(jīng)常出現(xiàn),題目源于《Python Cookbook》這本書第一章的第10個問題,有很多面試題其實都是這本書上的原題,所以建議大家有時間的話好好研讀一下這本書。
def dedup(items): no_dup_items = [] seen = set() for item in items: if item not in seen: no_dup_items.append(item) seen.add(item) return no_dup_items
當(dāng)然,也可以像《Python Cookbook》書上的代碼那樣,把上面的函數(shù)改造成一個生成器。
def dedup(items): seen = set() for item in items: if item not in seen: yield item seen.add(item) 擴展:
由于Python中的集合底層使用哈希存儲,所以集合的in和not in成員運算在性能上遠(yuǎn)遠(yuǎn)優(yōu)于列表,所以上面的代碼我們使用了集合來保存已經(jīng)出現(xiàn)過的元素。
集合中的元素必須是hashable對象,因此上面的代碼在列表元素不是hashable對象時會失效,要解決這個問題可以給函數(shù)增加一個參數(shù),該參數(shù)可以設(shè)計為返回哈希碼或hashable對象的函數(shù)。