1. 程式人生 > >python 列表展開,適用於N層巢狀的列表

python 列表展開,適用於N層巢狀的列表

我的簡書地址: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'] ] ] ] ] ] ]

demo for the flatten() function

完全沒毛病,支援各種資料型別。