python - 控制嵌套列表/字符串的递归

Question

假设我有以下输入:

items = [1, 2, [3, 4], (5, 6), 'ciao', range(3), (i for i in range(3, 6))]

我想对items执行一些递归操作。

为了简单起见，假设我想要展平项目(但也可以是其他任何东西)，一种方法是:

def flatten(items, shallow=(str, bytes, bytearray)):
    for item in items:
        if isinstance(item, shallow):
            yield item
        else:
            try:
                yield from flatten(item)
            except TypeError:
                yield item

这会产生:

print(list(flatten(items)))
[1, 2, 3, 4, 5, 6, 'ciao', 0, 1, 2, 3, 4, 5]

现在我如何修改 flatten() 以便我可以生成以下内容(对于任意嵌套级别)？

print(list(flatten(items)))
[1, 2, 3, 4, 5, 6, 'c', 'i', 'a', 'o', 0, 1, 2, 3, 4, 5]

Answer 1

只需在浅检查旁边添加一个长度检查:

if isinstance(item, shallow) and len(item) == 1: