這篇文章給大家分享的是有關怎么寫python3展開列表的代碼的內容。小編覺得挺實用的，因此分享給大家做個參考。一起跟隨小編過來看看吧。

展開列表

該方法將通過遞歸的方式將列表的嵌套展開為單個列表。

def spread(arg):
   ret = []
   for i in arg:
       if isinstance(i, list):
           ret.extend(i)
       else:
           ret.append(i)
   return ret

def deep_flatten(lst):
   result = []
   result.extend(
       spread(list(map(lambda x: deep_flatten(x) if type(x) == list else x, lst))))
   return result


deep_flatten([1, [2], [[3], 4], 5]) # [1,2,3,4,5]

拓展：列表的差

該方法將返回第一個列表的元素，其不在第二個列表內。如果同時要反饋第二個列表獨有的元素，還需要加一set_b.difference(set_a)。

def difference(a, b):
   set_a = set(a)
   set_b = set(b)
   comparison = set_a.difference(set_b)
   return list(comparison)


difference([1,2,3], [1,2,4]) # [3]

感謝各位的閱讀！關于怎么寫python3展開列表的代碼就分享到這里了，希望以上內容可以對大家有一定的幫助，讓大家可以學到更多知識。如果覺得文章不錯，可以把它分享出去讓更多的人看到吧！