Python 如何展开嵌套的序列
作者:David Beazley 时间:2022-10-12 03:15:37
问题
你想将一个多层嵌套的序列展开成一个单层列表
解决方案
可以写一个包含 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)
在上面代码中, isinstance(x, Iterable)
检查某个元素是否是可迭代的。 如果是的话, yield from
就会返回所有子例程的值。最终返回结果就是一个没有嵌套的简单序列了。
额外的参数 ignore_types
和检测语句 isinstance(x, ignore_types)
用来将字符串和字节排除在可迭代对象外,防止将它们再展开成单个的字符。 这样的话字符串数组就能最终返回我们所期望的结果了。比如:
>>> items = ['Dave', 'Paula', ['Thomas', 'Lewis']]
>>> for x in flatten(items):
... print(x)
...
Dave
Paula
Thomas
Lewis
>>>
讨论
语句 yield from
在你想在生成器中调用其他生成器作为子例程的时候非常有用。 如果你不使用它的话,那么就必须写额外的 for
循环了。比如:
def flatten(items, ignore_types=(str, bytes)):
for x in items:
if isinstance(x, Iterable) and not isinstance(x, ignore_types):
for i in flatten(x):
yield i
else:
yield x
尽管只改了一点点,但是 yield from
语句看上去感觉更好,并且也使得代码更简洁清爽。
之前提到的对于字符串和字节的额外检查是为了防止将它们再展开成单个字符。 如果还有其他你不想展开的类型,修改参数 ignore_types
即可。
最后要注意的一点是, yield from
在涉及到基于协程和生成器的并发编程中扮演着更加重要的角色。
来源:https://python3-cookbook.readthedocs.io/zh_CN/latest/c04/p14_flattening_nested_sequence.html
标签:Python,嵌套,序列
0
投稿
猜你喜欢
Pandas 同元素多列去重的实例
2023-02-09 21:03:32
Python 异步如何使用等待有时间限制协程
2022-02-12 09:29:28
Javascript 文件夹选择框的两种解决方案
2024-04-16 09:50:46
通过SqlCmd执行超大SQL文件的方法
2024-01-21 00:35:50
Python远程开发环境部署与调试过程图解
2023-06-07 08:10:40
OpenCV中VideoCapture类的使用详解
2022-10-26 18:38:15
python使用openpyxl库读写Excel表格的方法(增删改查操作)
2021-11-29 01:22:43
浅谈python中常用的excel模块库
2021-04-20 11:29:41
TypeScript学习之强制类型的转换
2024-04-25 13:08:07
python3利用Dlib19.7实现人脸68个特征点标定
2021-05-05 19:13:06
element-ui表格列金额显示两位小数的方法
2024-04-26 17:41:10
Python : turtle色彩控制实例详解
2022-09-22 13:01:46
Javascript中的isNaN函数使用说明
2023-08-27 10:10:02
详解Python调试神器之PySnooper
2021-05-20 06:51:12
python里 super类的工作原理详解
2022-06-13 23:54:14
Django uwsgi Nginx 的生产环境部署详解
2023-10-21 14:55:30
python中Task封装协程的知识点总结
2022-10-25 13:25:02
MySQL分区表的最佳实践指南
2024-01-27 14:45:40
Mysql事务的隔离级别(脏读+幻读+可重复读)
2024-01-29 07:45:06
Python 完美解决 Import “模块“ could not be resolved ...的问题
2022-04-05 18:59:29