[英]Control recursion on nested lists / strings
假设我有以下输入:
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]
只需在浅检查旁边添加一个长度检查:
if isinstance(item, shallow) and len(item) == 1:
声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.