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

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: