python 展开嵌套的序列

转载于 2018-12-14 17:11:00 发布 · 139 阅读

0 ·

CC 4.0 BY-SA版权

原文链接：http://www.cnblogs.com/baxianhua/p/10120507.html

文章标签：

#python

本文介绍了一种使用Python的yieldfrom语句和递归生成器将多层嵌套序列转换为单层列表的方法。通过定义一个名为flatten的函数，可以轻松地处理包括列表在内的各种可迭代对象，同时排除字符串和字节类型。

将一个多层嵌套的序列展开成一个单层列表

可以写一个包含yield from 语句的递归生成器来轻松解决这个问题。

from collections import Iterable
def flatten(items, ignore_types=(str, bytes)):
for x in items:
if isinstance(x, Iterable) and not isinstance(x, ignore_types):
yield from flatten(x)
else:
yield x
items = [1, 2, [3, 4, [5, 6], 7], 8]
# Produces 1 2 3 4 5 6 7 8
for x in flatten(items):
print(x)

bb=list(flatten(items))
print(bb)