用于列表或子列表中的项目

Question

我有一个列表，其中可能包含也可能不包含其他子列表。 我需要为列表中的每个项目以及列表中任何子列表中的每个项目运行代码块。 这是我的for语句现在的样子：

for item in mylist:
    ...

什么是重写所以最有效的方法item是从来没有的子列表中的一个，所以该代码将再次在所有子表的运行每一个项目，除了列表中的每个项目？

Answer 1

有两种通用的方法可以递归进行：

def unroll_recursive(lst):
    for el in lst:
        if isinstance(el, list):
            yield from unroll_recursive(el)
        else:
            yield el

或反复

from collections import deque
def unroll_iterative(lst)
    q = deque(lst)
    while q:
        el = q.popleft()
        if isinstance(el, list):
            q.extend(el)
        else:
            yield el