python 多層嵌套列表 轉成一維列表

如以下的嵌套列表:

[1,[2,3],[4,[5,6]]]

因爲不知道嵌套了多少層,常規的方式是寫一個遞歸函數,對這個函數進行遞歸來生成一維列表。

後來發現,在python的compiler模塊中,提供了類似功能的函數:

>>> from compiler.ast import flatten
>>> flatten(a)
[1, 2, 3, 4, 5, 6]

這樣子就簡單多了

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章