python 列表展開,適用於N層巢狀的列表
阿新 • • 發佈:2019-02-01
我的簡書地址:http://www.jianshu.com/p/8215f455e4e5
網路上公佈了很多列表巢狀展開的語法,這裡就不一一列舉了,方法大致包含 lambda語句, itertools. 這些工具存在問題,就是不能對N(N>3)的列表進行展開。曾經最喜歡用的神器 from compiler.ast import flatten 可以解決這個問題,但是現在python3 不支援這個包了。 試來試去,索性不如自己寫一個。
def flatten(input_list):
output_list = []
while True:
if input_list == []:
break
for index, i in enumerate(input_list):
if type(i)== list:
input_list = i + input_list[index+1:]
break
else:
output_list.append(i)
input_list.pop(index)
break
return output_list
程式碼的原理比較簡單,虛擬碼是:
1. 遍歷原始列表
2. 對於列表中的當前元素
2.1 如果它是非列表元素,存入輸出列表。
2.2 如果它是列表元素:進入該列表,執行2.1
一句話說清楚就是“遍歷原始列表,對於列表內的元素,如果非列表就存入輸出列表中,如果是列表元素,則進入改列表元素並檢視列表內元素型別”
測試一下:
a = [ 1, 2, [3, 4, [5, 6, [7, 8, [9, [10, ['end'] ] ] ] ] ] ]
完全沒毛病,支援各種資料型別。