0

0

如何为 Python 生成器对象自定义高效的 __contains__ 行为

花韻仙語

花韻仙語

发布时间:2026-02-13 14:04:21

|

947人浏览过

|

来源于php中文网

原创

如何为 Python 生成器对象自定义高效的 __contains__ 行为

python 原生生成器不支持自定义 `__contains__`,无法避免遍历查找;本文介绍通过可迭代类模拟生成器行为,在保持惰性迭代优势的同时,实现 o(1) 或逻辑判断式的成员检测。

在 Python 中,由函数 yield 返回的原生生成器对象(generator 类型)是不可变的内置类型,其 __contains__ 方法已固定为线性遍历实现——即调用 in 操作符时,会从头开始逐个 next() 直到匹配或耗尽。正如示例中 9999 in myGenerator() 所示,即使目标值靠前,实际仍需执行大量无谓迭代(如打印 iterating 0 到 iterating 9998),完全丧失生成器“按需计算”的核心价值。

因此,直接为原生 generator 对象注入或重写 __contains__ 是不可能的——它既没有公开的构造接口,也不允许动态设置特殊方法。真正的解决方案是:放弃原生生成器,改用自定义可迭代类(Iterable + Iterator 协议)来模拟其行为,并自主控制 __contains__ 的逻辑

以下是一个专业、轻量且符合 Python 惯例的实现:

Voiceflow
Voiceflow

Voiceflow 是一个AI驱动的聊天机器人构建平台,可以帮您设计、开发和发布聊天机器人。

下载
class MySequence:
    def __init__(self, max_val=1000000):
        self.max_val = max_val

    def __iter__(self):
        return MySequenceIterator(self.max_val)

    def __contains__(self, value):
        # ✅ 高效判断:无需遍历,仅做数学/逻辑校验
        return isinstance(value, int) and 0 <= value < self.max_val

class MySequenceIterator:
    def __init__(self, max_val):
        self.max_val = max_val
        self.current = -1

    def __iter__(self):
        return self

    def __next__(self):
        self.current += 1
        if self.current >= self.max_val:
            raise StopIteration
        print(f"iterating {self.current}")
        return self.current

使用方式与原生生成器高度一致:

立即学习Python免费学习笔记(深入)”;

seq = MySequence(1000000)

# ✅ 成员检测:O(1),立即返回 True/False
print(9999 in seq)      # 输出: True(无任何 iter 输出)

# ✅ 惰性迭代:仍按需生成,支持 for 循环、next() 等
for i in seq:
    if i > 3:
        break
# 输出:
# iterating 0
# iterating 1
# iterating 2
# iterating 3
# iterating 4
⚠️ 关键注意事项:不要在 __contains__ 中触发实际迭代:这是常见误区。正确做法是基于序列的数学规律(如范围、公式、哈希预判等)设计逻辑判断。分离迭代器状态:将 __iter__ 返回独立的 Iterator 实例(如 MySequenceIterator),确保多次迭代互不干扰,符合 Python 迭代器协议。避免在类体中定义实例变量(如原答案中的 i = -1):这会导致所有实例共享同一状态,引发严重 bug;务必在 __init__ 或迭代器中初始化。性能权衡:若业务逻辑本身无法避免遍历(如模糊匹配、外部 API 查询),则自定义 __contains__ 仍需遍历——此时应明确文档说明,或考虑缓存策略(如 functools.lru_cache + tuple() 化首 N 项)。

总结而言,当需要高效 in 检测时,生成器不是最佳选择;而通过标准迭代器协议构建的类,既能保留惰性求值语义,又能赋予你对 __contains__ 的完全控制权——这是一种更健壮、更可维护、也更符合 Python 设计哲学的工程实践。

热门AI工具

更多
DeepSeek
DeepSeek

幻方量化公司旗下的开源大模型平台

豆包大模型
豆包大模型

字节跳动自主研发的一系列大型语言模型

通义千问
通义千问

阿里巴巴推出的全能AI助手

腾讯元宝
腾讯元宝

腾讯混元平台推出的AI助手

文心一言
文心一言

文心一言是百度开发的AI聊天机器人,通过对话可以生成各种形式的内容。

讯飞写作
讯飞写作

基于讯飞星火大模型的AI写作工具,可以快速生成新闻稿件、品宣文案、工作总结、心得体会等各种文文稿

即梦AI
即梦AI

一站式AI创作平台,免费AI图片和视频生成。

ChatGPT
ChatGPT

最最强大的AI聊天机器人程序,ChatGPT不单是聊天机器人,还能进行撰写邮件、视频脚本、文案、翻译、代码等任务。

相关专题

更多
硬盘接口类型介绍
硬盘接口类型介绍

硬盘接口类型有IDE、SATA、SCSI、Fibre Channel、USB、eSATA、mSATA、PCIe等等。详细介绍:1、IDE接口是一种并行接口,主要用于连接硬盘和光驱等设备,它主要有两种类型:ATA和ATAPI,IDE接口已经逐渐被SATA接口;2、SATA接口是一种串行接口,相较于IDE接口,它具有更高的传输速度、更低的功耗和更小的体积;3、SCSI接口等等。

1440

2023.10.19

PHP接口编写教程
PHP接口编写教程

本专题整合了PHP接口编写教程,阅读专题下面的文章了解更多详细内容。

340

2025.10.17

php8.4实现接口限流的教程
php8.4实现接口限流的教程

PHP8.4本身不内置限流功能,需借助Redis(令牌桶)或Swoole(漏桶)实现;文件锁因I/O瓶颈、无跨机共享、秒级精度等缺陷不适用高并发场景。本专题为大家提供相关的文章、下载、课程内容,供大家免费下载体验。

2210

2025.12.29

java接口相关教程
java接口相关教程

本专题整合了java接口相关内容,阅读专题下面的文章了解更多详细内容。

36

2026.01.19

pixiv网页版官网登录与阅读指南_pixiv官网直达入口与在线访问方法
pixiv网页版官网登录与阅读指南_pixiv官网直达入口与在线访问方法

本专题系统整理pixiv网页版官网入口及登录访问方式,涵盖官网登录页面直达路径、在线阅读入口及快速进入方法说明,帮助用户高效找到pixiv官方网站,实现便捷、安全的网页端浏览与账号登录体验。

23

2026.02.13

微博网页版主页入口与登录指南_官方网页端快速访问方法
微博网页版主页入口与登录指南_官方网页端快速访问方法

本专题系统整理微博网页版官方入口及网页端登录方式,涵盖首页直达地址、账号登录流程与常见访问问题说明,帮助用户快速找到微博官网主页,实现便捷、安全的网页端登录与内容浏览体验。

11

2026.02.13

Flutter跨平台开发与状态管理实战
Flutter跨平台开发与状态管理实战

本专题围绕Flutter框架展开,系统讲解跨平台UI构建原理与状态管理方案。内容涵盖Widget生命周期、路由管理、Provider与Bloc状态管理模式、网络请求封装及性能优化技巧。通过实战项目演示,帮助开发者构建流畅、可维护的跨平台移动应用。

7

2026.02.13

TypeScript工程化开发与Vite构建优化实践
TypeScript工程化开发与Vite构建优化实践

本专题面向前端开发者,深入讲解 TypeScript 类型系统与大型项目结构设计方法,并结合 Vite 构建工具优化前端工程化流程。内容包括模块化设计、类型声明管理、代码分割、热更新原理以及构建性能调优。通过完整项目示例,帮助开发者提升代码可维护性与开发效率。

8

2026.02.13

Redis高可用架构与分布式缓存实战
Redis高可用架构与分布式缓存实战

本专题围绕 Redis 在高并发系统中的应用展开,系统讲解主从复制、哨兵机制、Cluster 集群模式及数据分片原理。内容涵盖缓存穿透与雪崩解决方案、分布式锁实现、热点数据优化及持久化策略。通过真实业务场景演示,帮助开发者构建高可用、可扩展的分布式缓存系统。

3

2026.02.13

热门下载

更多
网站特效
/
网站源码
/
网站素材
/
前端模板

精品课程

更多
相关推荐
/
热门推荐
/
最新课程
最新Python教程 从入门到精通
最新Python教程 从入门到精通

共4课时 | 22.4万人学习

Django 教程
Django 教程

共28课时 | 4.2万人学习

SciPy 教程
SciPy 教程

共10课时 | 1.5万人学习

关于我们 免责申明 举报中心 意见反馈 讲师合作 广告合作 最新更新
php中文网:公益在线php培训,帮助PHP学习者快速成长!
关注服务号 技术交流群
PHP中文网订阅号
每天精选资源文章推送

Copyright 2014-2026 https://www.php.cn/ All Rights Reserved | php.cn | 湘ICP备2023035733号