百度360必应搜狗淘宝本站头条
当前位置:网站首页 > 技术资源 > 正文

Python生成器入门:用“按需生产”思维处理海量数据

off999 2025-05-23 19:15 12 浏览 0 评论

想象你要制作1000个蛋糕:

  • 列表的做法:先把1000个蛋糕全做好堆在厨房里,占满空间还容易变质
  • 生成器的做法:有人下单时才做一个蛋糕,厨房永远只有1个蛋糕,不占空间还新鲜

这就是Python生成器的核心逻辑——按需生成数据,绝不提前囤积。本文将用漫画式思维和极简代码,带新手彻底理解这个“内存救星”。

一、为什么列表处理大数据会“卡爆”?

案例:生成1亿个数字

# 用列表存储1亿个数字(灾难现场!)
big_list = [i for i in range(100000000)]  # 电脑可能直接卡死

问题根源

  • 列表像一个“超级仓库”,必须把所有数据一次性存进去
  • 数据量越大,仓库越占空间,小电脑根本扛不住!

二、生成器:数据界的“自动贩卖机”

生成器就像一台自动贩卖机:

  • 你按下按钮(调用next())
  • 贩卖机现做一份零食(生成一个数据)
  • 绝不提前生产堆在机器里

1. 生成器表达式:贩卖机的简易版

语法

(数据生成规则 for 原材料 in 供应列表)  # 用()代替列表的[]

示例:生成1-5的平方贩卖机

square_gen = (x**2 for x in range(1, 6))  # 先造一台贩卖机,不生产任何数据

如何买零食?

print(next(square_gen))  # 按下按钮:得到1(贩卖机生产第1个数据)
print(next(square_gen))  # 再按按钮:得到4(生产第2个数据)
# 每次调用next(),贩卖机才生产一个数据,内存始终只有1个数据!

2. 生成器函数:贩卖机的定制版

如果需要复杂的生产流程(如筛选、计算),可以用函数造贩卖机:
语法

def 贩卖机函数():
    for 原材料 in 供应列表:
        if 原材料符合条件:
            yield 加工后的产品  # yield是“出货口”,每次出一个产品后暂停

示例:生成100以内的偶数贩卖机

def even_gen():
    for x in range(100):
        if x % 2 == 0:
            yield x  # 出货口:给我一个偶数!

gen = even_gen()  # 造好贩卖机
print(next(gen))  # 0(第1次出货)
print(next(gen))  # 2(第2次出货)

三、生成器的超能力:内存占用永不爆炸

实验对比:列表vs生成器的内存消耗

数据量

列表占用内存(Python实测)

生成器占用内存

10个数字

约0.04KB

约0.0008KB

100万个数字

约40KB

约0.0008KB

1亿个数字

约4000KB(4MB)

约0.0008KB

原因

  • 列表:存1亿个数字 → 每个数字占4字节 → 总内存4亿字节=40MB
  • 生成器:只存“从0开始,每次+1”的规则 → 仅需几十字节记录规则

四、生成器的3个经典使用场景

场景1:处理超大文件(如10GB日志)

传统方法(错误示范)

with open('big.log', 'r') as f:
    lines = f.readlines()  # 把10GB文件全读进内存,电脑直接罢工!

生成器方法(正确做法)

def read_large_file(file_path):
    with open(file_path, 'r') as f:
        for line in f:  # 文件对象本身就是生成器,逐行读取
            yield line.strip()  # 每次只存1行数据,内存稳如老狗!

# 逐行处理:统计包含"ERROR"的行数
error_count = 0
for line in read_large_file('big.log'):
    if "ERROR" in line:
        error_count += 1

场景2:实时生成数据(如模拟传感器)

import time

def live_sensor():
    while True:
        yield time.strftime("%H:%M:%S")  # 无限生成当前时间
        time.sleep(1)  # 每秒生成1个数据

# 使用:打印实时时间,不存历史数据
sensor = live_sensor()
print(next(sensor))  # 15:30:01
print(next(sensor))  # 15:30:02
# 内存始终只存最新1个时间,绝不积压!

场景3:数据流水线(过滤→转换→分析)

# 生成器链:数据像流水一样逐个处理
def generate_data():
    yield 1; yield 2; yield 3; yield 4  # 生成原始数据

def filter_even(data):
    for x in data:
        if x % 2 == 0:
            yield x  # 过滤偶数

def convert_to_str(data):
    for x in data:
        yield f"数字{x}"  # 转换格式

# 流水线:生成→过滤→转换
pipeline = convert_to_str(filter_even(generate_data()))
for item in pipeline:
    print(item)  # 输出:"数字2", "数字4"

五、新手必看!生成器的3个“不能做”

不能做1:用索引访问生成器

gen = (x for x in range(5))
print(gen[2])  #  报错!生成器不能像列表一样用索引
# 正确做法:用for循环遍历或next()逐个获取

不能做2:重复使用耗尽的生成器

gen = (x for x in range(3))
print(next(gen))  # 0
print(next(gen))  # 1
print(next(gen))  # 2
print(next(gen))  #  报错!生成器已空
# 解决办法:重新创建生成器对象 gen = (x for x in range(3))

不能做3:用生成器存储需要反复访问的数据

#  错误场景:需要多次遍历数据
gen = (x for x in range(5))
print(sum(gen))  # 10(第一次遍历,生成器耗尽)
print(sum(gen))  # 0(第二次遍历,没有数据了!)

# 正确做法:用列表存储需要反复使用的数据
lst = [x for x in range(5)]
print(sum(lst))  # 10
print(sum(lst))  # 10(列表数据一直都在)

六、实战练习:用生成器计算10000以内的质数

需求:生成10000以内的所有质数,要求内存占用小于1MB
步骤1:定义质数判断函数

def is_prime(n):
    if n < 2:
        return False
    for i in range(2, int(n**0.5)+1):
        if n % i == 0:
            return False
    return True

步骤2:用生成器函数生成质数

def prime_generator(max_num):
    for n in range(2, max_num+1):
        if is_prime(n):
            yield n  # 是质数就出货!

步骤3:遍历生成器并统计

primes = prime_generator(10000)
print(f"10000以内的质数有:{list(primes)}")
# 内存占用:生成器全程仅存储当前判断的数字,远小于1MB!

总结:生成器的“三句口诀”

  1. 列表是仓库,生成器是流水线:列表存所有数据,生成器逐个造数据
  2. 用for循环替代next():for x in 生成器: 比手动调用next()更简单
  3. 大数据用生成器,小数据用列表:处理几万个数据用列表,百万级数据用生成器

给新手的练习建议

  1. 用生成器表达式生成1-100的奇数序列
  2. 尝试用生成器函数实现斐波那契数列(提示:用a, b = b, a+b)
  3. 思考:如何用生成器处理CSV文件的逐行解析?

生成器是Python中“聪明处理数据”的关键工具,学会它能让你在处理大数据时游刃有余。下次我们将深入迭代器原理,揭开生成器背后的神秘面纱,记得关注哦!

相关推荐

推荐一款Python的GUI可视化工具(python 可视化工具)

在Python基础语法学习完成后,进一步开发应用界面时,就需要涉及到GUI了,GUI全称是图形用户界面(GraphicalUserInterface,又称图形用户接口),采用图形方式显示的计算机操...

教你用Python绘制谷歌浏览器的3种图标

前两天在浏览matplotlib官方网站时,笔者无意中看到一个挺有意思的图片,就是用matplotlib制作的火狐浏览器的logo,也就是下面这个东东(网页地址是https://matplotlib....

小白学Python笔记:第二章 Python安装

Windows操作系统的python安装:Python提供Windows、Linux/UNIX、macOS及其他操作系统的安装包版本,结合自己的使用情况,此处仅记录windows操作系统的python...

Python程序开发之简单小程序实例(9)利用Canvas绘制图形和文字

Python程序开发之简单小程序实例(9)利用Canvas绘制图形和文字一、项目功能利用Tkinter组件中的Canvas绘制图形和文字。二、项目分析要在窗体中绘制图形和文字,需先导入Tkinter组...

一文吃透Python虚拟环境(python虚拟环境安装和配置)

摘要在Python开发中,虚拟环境是一种重要的工具,用于隔离不同项目的依赖关系和环境配置。本文将基于windows平台介绍四种常用的Python虚拟环境创建工具:venv、virtualenv、pip...

小白也可以玩的Python爬虫库,收藏一下

最近,微软开源了一个项目叫「playwright-python」,作为一个兴起项目,出现后受到了大家热烈的欢迎,那它到底是什么样的存在呢?今天为你介绍一下这个传说中的小白神器。Playwright是...

python环境安装+配置教程(python安装后怎么配置环境变量)

安装python双击以下软件:弹出一下窗口需选择一些特定的选项默认选项不需要更改,点击next勾选以上选项,点击install进度条安装完毕即可。到以下界面,证明安装成功。接下来安装库文件返回电脑桌面...

colorama,一个超好用的 Python 库!

大家好,今天为大家分享一个超好用的Python库-colorama。Github地址:https://github.com/tartley/coloramaPythoncolorama库是一...

python制作仪表盘图(python绘制仪表盘)

今天教大家用pyecharts画仪表盘仪表盘(Gauge)是一种拟物化的图表,刻度表示度量,指针表示维度,指针角度表示数值。仪表盘图表就像汽车的速度表一样,有一个圆形的表盘及相应的刻度,有一个指针...

总结90条写Python程序的建议(python写作)

  1.首先  建议1、理解Pythonic概念—-详见Python中的《Python之禅》  建议2、编写Pythonic代码  (1)避免不规范代码,比如只用大小写区分变量、使用容易...

[oeasy]python0137_相加运算_python之禅_import_this_显式转化

变量类型相加运算回忆上次内容上次讲了是从键盘输入变量input函数可以有提示字符串需要有具体的变量接收输入的字符串输入单个变量没有问题但是输入两个变量之后一相加就非常离谱添加图片注释,不超过1...

Python入门学习记录之一:变量(python中变量的规则)

写这个,主要是对自己学习python知识的一个总结,也是加深自己的印象。变量(英文:variable),也叫标识符。在python中,变量的命名规则有以下三点:>变量名只能包含字母、数字和下划线...

掌握Python的&quot;魔法&quot;:特殊方法与属性完全指南

在Python的世界里,以双下划线开头和结尾的"魔法成员"(如__init__、__str__)是面向对象编程的核心。它们赋予开发者定制类行为的超能力,让自定义对象像内置类型一样优雅工...

11个Python技巧 不Pythonic 实用大于纯粹

虽然Python有一套强大的设计哲学(体现在“Python之禅”中),但总有一些情况需要我们“打破规则”来解决特定问题。这触及了Python哲学中一个非常核心的理念:“实用主义胜于纯粹主义”...

Python 从入门到精通 第三课 诗意的Python之禅

导言:Python之禅,英文名是TheZenOfPython。最早由TimPeters在Python邮件列表中发表,它包含了影响Python编程语言设计的20条软件编写原则。它作为复活节彩蛋...

取消回复欢迎 发表评论: