从父列表中删除嵌套列表

时间:2017-01-07 01:26:22

标签: python recursion iteration

我们说我有以下列表data,我想列出flat_data列表:

data = [[1, 2], 3, 4, [5, [6, 7]]]
flat_data = [1, 2, 3, 4, 5, 6, 7]

最好的方法是什么?

这是我到目前为止所做的事情,但理想情况下我想把所有东西都塞进一个功能中:

flat_data = []
def flatten_data(data):
    for i in data:
        if type(i) == list:
            flatten_data(i)
        else:
            flat_data.append(i)
    return flat_a

flatten_data(data)
# [1, 2, 3, 4, 5, 6, 7]

0 个答案:

没有答案