總第 116 篇文章,本文大約 1000 字,閱讀大約需要 3 分鐘
今天介紹和列表相關的兩個小技巧:
碾平列表(flatten list),也就是列表裏的元素也帶有列表的情況;
列表去重,保留原始順序和不保留順序的做法
1. 碾平列表
碾平列表(flatten list ),即當列表裏面嵌套列表,如何將這些子列表給取出來,得到一個不包含子列表的列表,示例如下:
list1 = [1, [2, [3,4]], 5]
=>new_list = [1, 2, 3, 4, 5]
這裏介紹 3 種方法,分別如下。
方法1:利用遞歸的思想,代碼如下:
list1 = [1, [2, [3,4]], 5]
res = []
def fun(s):
for i in s:
if isinstance(i, list):
fun(i)
else:
res.append(i)
fun(list1)
print(res)
接着是兩種比較高級的寫法,用 lambda
實現一個匿名函數
方法2:
flat = lambda L: sum(map(flat, L), []) if isinstance(L, list) else [L]
print(flat(list1))
方法3:
a = [1, 2, [3, 4], [[5, 6], [7, 8]]]
flatten = lambda x: [y for l in x for y in flatten(l)] if type(x) is list else [x]
print(flatten(a))
2. 列表去重
列表去重可能會破壞原有的順序,所以下面分別介紹保留順序和不保留順序的做法。
去重,但改變順序
去重但改變順序,兩種方法
方法1 就是利用 set
進行去重
l1 = ['b','c','d','b','c','a','a']
l2 = list(set(l1))
print l2
方法2 是利用字典的鍵不重複的特性,將列表的元素作爲一個字典的鍵,然後返回這個字典的所有鍵,即可實現去重的操作。
l1 = ['b','c','d','b','c','a','a']
l2 = {}.fromkeys(l1).keys()
print l2
去重,不改變順序
利用 sorted
和 set
方法實現去重並保留原始順序,這裏 sorted
指定排序的規則就是按照原列表的索引順序
l1 = ['b','c','d','b','c','a','a']
l2 = sorted(set(l1),key=l1.index)
print l2
歡迎關注我的微信公衆號--算法猿的成長,或者掃描下方的二維碼,大家一起交流,學習和進步!
如果覺得不錯,在看、轉發就是對小編的一個支持!