1

可能重复:
在 Python 中展平(不规则)列表列表

示例 1:

假设我有一个列表:[[1,2],[3,4]]。我可以使用两个 for 循环来打印:1、2、3、4。

示例 2:

所以,现在让我们假设我得到了一个输出,但我不知道 list1 中有多少嵌套列表:

list1 = [1, [1, 2, [3, 5, 6,[ .. ], ..., ] ] ] ] ]

所以,我的问题是如何以与第一个示例相同的格式打印出每个单独的数字。我现在正在处理一些结果,它给了我嵌套列表,但是函数的不同输入会给我不同数量的嵌套列表。

我能想到的就是这样做,但是我不知道在isinstance部分之后要做什么:

c = 0
for i in list1:
   while c < len(list1):
         if isinstance(i, list):

         else:
              print i
         c += 1

谢谢

第一次编辑

如果还有一种方法可以将所有嵌套列表解构为一个对我也有用的列表,但我很想知道这两个问题的答案。

4

3 回答 3

4

itertools 文档有一些很好的迭代列表等示例,因此当面对这样的任务时,它总是一个很好的起点。

我建议使用生成器,它可以避免创建多个级别的列表:

def flatten_all(iterable):
    for elem in iterable:
        if not isinstance(elem, list):
            yield elem
        else:
            for x in flatten_all(elem):
                yield x
            # in Python 3.3 just: yield from flatten_all(elem)

应用:

for x in flatten_all([1, [2, [3]]]):
    print(x)

# or if you need a list:
my_lst = list(flatten_all([1, [2, [3]]])
assert my_lst == [1, 2, 3]

编辑:非递归线性版本

def flatten_all(iterable):
    stack = [iter(iterable)]
    while stack:
        try:
            elem  = stack[-1].next()
            if not isinstance(elem, list):
                yield elem
            else:
                stack.append(iter(elem))
        except StopIteration:
            stack.pop()
于 2012-11-16T10:33:34.323 回答
1

您可以尝试这样的事情(代码使您的列表平放在一行中,以便您稍后打印):

def make_flat(arr):
    res = []
    for l in arr:
        if isinstance(l, list):# or isinstance(l, collections.Iterable)
            res.extend(make_flat(l)) 
        else:
            res.append(l)
    return res

flat = make_flat(list1)
for x in flat:
    print x

或者:

def make_flat(arr):
    return sum(map(lambda a: make_flat(a) if isinstance(a,(list)) else [a],arr),[])
于 2012-11-16T10:26:05.283 回答
1

这是一个使用递归的例子:

list1 = [1, [2,3,4], [5, 6, 7, [8, 9]]]

def print_list(l):
    for e in l:
        if type(e) == list:
            print_list(e)
        else:
            print e

print_list(list1)
于 2012-11-16T10:27:38.323 回答