Python 如何展开嵌套的序列
问题
你想将一个多层嵌套的序列展开成一个单层列表
解决方案
可以写一个包含yieldfrom语句的递归生成器来轻松解决这个问题。比如:
fromcollectionsimportIterable defflatten(items,ignore_types=(str,bytes)): forxinitems: ifisinstance(x,Iterable)andnotisinstance(x,ignore_types): yieldfromflatten(x) else: yieldx items=[1,2,[3,4,[5,6],7],8] #Produces12345678 forxinflatten(items): print(x)
在上面代码中,isinstance(x,Iterable)检查某个元素是否是可迭代的。如果是的话,yieldfrom就会返回所有子例程的值。最终返回结果就是一个没有嵌套的简单序列了。
额外的参数ignore_types和检测语句isinstance(x,ignore_types)用来将字符串和字节排除在可迭代对象外,防止将它们再展开成单个的字符。这样的话字符串数组就能最终返回我们所期望的结果了。比如:
>>>items=['Dave','Paula',['Thomas','Lewis']] >>>forxinflatten(items): ...print(x) ... Dave Paula Thomas Lewis >>>
讨论
语句yieldfrom在你想在生成器中调用其他生成器作为子例程的时候非常有用。如果你不使用它的话,那么就必须写额外的for循环了。比如:
defflatten(items,ignore_types=(str,bytes)): forxinitems: ifisinstance(x,Iterable)andnotisinstance(x,ignore_types): foriinflatten(x): yieldi else: yieldx
尽管只改了一点点,但是yieldfrom语句看上去感觉更好,并且也使得代码更简洁清爽。
之前提到的对于字符串和字节的额外检查是为了防止将它们再展开成单个字符。如果还有其他你不想展开的类型,修改参数ignore_types即可。
最后要注意的一点是,yieldfrom在涉及到基于协程和生成器的并发编程中扮演着更加重要的角色。
以上就是Python如何展开嵌套的序列的详细内容,更多关于Python展开嵌套的序列的资料请关注毛票票其它相关文章!