从里到外迭代嵌套列表

时间:2013-02-19 14:52:46

标签: python python-3.x

我有以下python嵌套列表结构:

test = ['a', ['c', ['e'], 'd'], 'b']

或相同,只是格式化:

test = [
    'a', 
        [
            'c', 
                [
                    'e'
                ], 
             'd'
        ], 
    'b'
]

我想知道迭代完整列表的最佳方法是什么,从最里面的嵌套列表对象('e')开始到最外面的列表('a',[...],'b')按相反的顺序。对反向(测试)的调用只是没有嵌套列表的技巧。它应该能够在迭代的每个深度调用回调函数。

迭代看起来应该是这样的([xx] ==来自先前调用的回调的计算值):

1st e --> callback(e)
2nd c [e] d --> callback(c [e] d)
3rd a [c e d] b --> callback(a [c e d] b)

希望这能解释我的问题&谢谢你的帮助

2 个答案:

答案 0 :(得分:6)

我建议的一个可能的解决方案是

>>> def foo(test):
    queue = []
    try:
        while True:
            queue.append(test)
            test = test[1]
    except IndexError:
        for e in reversed(queue):
            yield e


>>> data = foo(test)
>>> next(data)
['e']
>>> next(data)
['c', ['e'], 'd']
>>> next(data)
['a', ['c', ['e'], 'd'], 'b']
>>> next(data)

Traceback (most recent call last):
  File "<pyshell#753>", line 1, in <module>
    next(data)
StopIteration
>>> 

工作原理

  1. 遍历深度优先,并推送队列中的元素
  2. 循环反转队列并生成元素

答案 1 :(得分:6)

您正在寻找的是结构的后序遍历

def traverse(l):
    for x in l:
        if isinstance(x, list):
            traverse(x)
    callback(l)

如果将callback定义为print,我们就会

['e']
['c', ['e'], 'd']
['a', ['c', ['e'], 'd'], 'b']