控制嵌套列表/字符串的遞歸

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: