告别卡顿!用Python生成器轻松处理海量数据的秘籍

不知道大家有没有处理过海量数据,就是数据量非常庞大的时候,如何处理?说实在的在实际项目中确实是没有处理过海量数据,但是没事儿的时候我们可以在本地模拟海量数据来玩玩儿的,那么处理海量数据用什么方式呢?用readlines()读取?还是用生成器和迭代器呢?恐怕大家用的应该也都是生成器和迭代器吧!要不然程序卡死不说,有可能把电脑搞崩溃呦!

那么生成器和迭代器怎么玩儿呢?咱们继续往下走着看着。

从生活场景理解迭代器

想象一下你去吃自助餐,不是一次性把所有的食物都端到你桌上那得多大的桌子啊!,而是想吃的时候去取一点。迭代器就是这样的工作方式。(PS:有的人可能就是一次性端上来,个例个例,比喻不是很恰当)

在Python中,任何实现了__iter__()__next__()方法的对象都是迭代器。它不会一次性把所有数据加载到内存中,而是按需"生成"每个元素。

Python中很多内置对象都是可迭代的:列表、元组、字符串、字典,甚至文件对象。这就是为什么你能用for循环遍历它们的原因。

生成器:更优雅的迭代器

写迭代器需要定义类并实现两个方法,有点麻烦。于是Python提供了更简单的工具------生成器。

生成器有两种创建方式:

1. 生成器函数:使用def定义,但用yield代替return

python 复制代码
def num_generator(num):
    for i in range(num):
        print(f"生成数字是:{i}")
        yield i


# 使用创建的生成器
gen_a = num_generator(5)

# 逐个获取值
for num in gen_a:
    print(f"接收到:{num}")

2. 生成器表达式:类似列表推导式,但用圆括号

python 复制代码
print("列表推导式VS生成器表达式:")

# 列表推导式-立即创建所有元素
list_comp = [x ** 2 for x in range(100000)]
# print(f"列表推导式:{list_comp}")
print(f"内存占用:{list_comp.__sizeof__()}字节")

# 生成器表达式-按需生成元素
gen_comp = (x ** 2 for x in range(100000))
# print(f"生成器表达式:{gen_comp}")
print(f"内存占用:{gen_comp.__sizeof__()}字节")

看看海量数据时,这个内存占用大不大?害怕不害怕?

这个时候问题就来了:为什么生成器能高效处理大数据?为什么呢?

为什么生成器能高效处理大数据?

这个主要就是生成器的一个执行机制:

  • 普通函数:一旦调用,就从头跑到尾(return),然后全部执行完毕,局部变量全部销毁。
  • 生成器函数 (使用 yield 的函数:如上面的第一幅截图):调用时返回一个生成器对象 ,但并不立即执行函数体。当第一次调用 next() 时,函数从开始处执行,直到遇到 yield 语句,暂停 并返回 yield 后的值。函数当前的整个局部状态(变量、指针等)都会被冻结保存。 下次再调用 next(),函数从上次暂停的 yield 语句后紧接着继续执行。

但是注意!

生成器只能遍历一次,如需重复使用需要重新创建,就是只能遍历一次,再次打印已空空如也

生成器的高级用法

生成器不仅能产出数据,还能通过send()方法接收数据:

python 复制代码
def interactive_generator():
    while True:
        received = yield  # 等待发送数据
        print(f"收到:{received}")

gen = interactive_generator()
next(gen)  # 启动生成器
gen.send("Hello")  # 输出:收到:Hello
gen.send("World")  # 输出:收到:World

这种方法在复杂的数据流水线处理中非常有用。

写到最后

生成器和迭代器是Python处理海量数据的"神器",它们通过惰性计算(需要时才生成值)节省了大量内存。 数据量不大时,用列表更简单直接; 处理GB级别以上数据时,一定要用生成器;管道式数据处理中,生成器可以串联形成高效处理链。

相关推荐
北斗落凡尘几秒前
LangGraph 入门实战(12)--使用MCP
后端·python·langchain
jdksjw21 分钟前
同步与异步、阻塞与非阻塞、多线程、协程超详细讲解(Python并发编程从入门到精通)
开发语言·python
Mike_Zhang1 小时前
使用python统计FreeSWITCH呼叫并发
python·freeswitch
SL-staff1 小时前
技术实践:HR如何用JVS-Logic可视化编排实现考勤数据同步(含节点配置与异常处理)
开发语言·python·低代码·钉钉·可视化编排·jvs-logic·hr技术
SomeB1oody2 小时前
【RustyML入门】6.0. 数学工具
开发语言·后端·机器学习·rust·教程
IT_陈寒2 小时前
为什么我的JavaScript闭包总是漏掉那个变量?
前端·人工智能·后端
zzzzzz3103 小时前
AI 助手最危险的不是报错,而是“看起来成功”:我给工作流加了四道保险
人工智能·python
用户8356290780513 小时前
如何使用 Python 将 PowerPoint 转换为 PDF 文档
后端·python
菜冻鱼3 小时前
Python-pytorch-高级技巧
开发语言·人工智能·pytorch·python·深度学习·神经网络·聚类
倔强的石头_3 小时前
从 8 行报错到一条故障链:我用蓝耘 MaaS 做了一个 AI 日志分析助手
后端