python 實現多層列表拆分成單層列表

有個多層列表：[1, 2, 3, 4, [5, 6, [7, 8]], ['a', 'b', [2, 4]]]，拆分成單層列表

使用內置方法

結果和原列表順序不同

def split(li):
    pop_index_list = []  # 用來存儲需要刪除元素的索引
    for ele in li:
        if isinstance(ele, list):
            pop_index_list.insert(0, li.index(ele))
            li.extend(ele)

    for i in pop_index_list:
        li.pop(i)

   
li = [1, 2, 3, 4, [5, 6, [7, 8]], ['a', 'b', [2, 4]]]
split(li)
print(li)
# 結果 [1, 2, 3, 4, 5, 6, 'a', 'b', 7, 8, 2, 4]

遞歸方式

def split(li:list, new_list=[]):
    for ele in li:
        if isinstance(ele, list):
            split(ele)
        else:
            new_list.append(ele)
    return new_list
li = [1, 2, 3, 4, [5, 6, [7, 8]], ['a', 'b', [2, 4]]]
print(split(li))
# 結果：[1, 2, 3, 4, 5, 6, 7, 8, 'a', 'b', 2, 4]