0

0

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

霞舞

霞舞

发布时间:2026-02-13 14:22:01

|

308人浏览过

|

来源于php中文网

原创

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

python 原生生成器对象不支持直接重写 `__contains__`,但可通过封装为可迭代类实现高效成员检测,避免遍历全部元素,兼顾惰性计算与逻辑优化。

在 Python 中,由函数返回的原生生成器(generator 类型)是不可变、一次性且无法动态注入特殊方法的对象。这意味着你无法像对自定义类那样直接为 myGenerator() 的返回值添加 __contains__ 方法——generator 类本身已固定实现,且其 __contains__ 底层就是线性遍历(调用 __next__ 直至匹配或耗尽),这正是你观察到“必须迭代到 9999 才返回 True”的根本原因。

要真正实现短路式、非遍历型的 in 检测(例如数学上可推导成员资格),正确路径是:放弃纯函数式生成器,改用自定义可迭代类(Iterable Class),显式实现 __iter__、__next__ 和 __contains__ 三个协议方法。

以下是一个专业、健壮的实现示例:

class BoundedCounter:
    def __init__(self, start=-1, stop=1000000):
        self.start = start
        self.stop = stop

    def __iter__(self):
        self._current = self.start
        return self

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

    def __contains__(self, value):
        # ✅ O(1) 成员判断:无需遍历,仅依赖数学约束
        return isinstance(value, int) and self.start < value < self.stop

# 使用示例
gen = BoundedCounter()

# 高效 in 检测(不触发任何迭代)
print(9999 in gen)      # True —— 瞬时返回,无打印输出
print(1000005 in gen)  # False —— 同样瞬时

# 正常迭代(惰性、按需执行)
for i, item in enumerate(gen):
    if i >= 4:  # 仅取前5个值
        break
# 输出:
# iterating 0
# iterating 1
# iterating 2
# iterating 3
# iterating 4

关键优势

NoCode
NoCode

美团推出的零代码应用生成平台

下载

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

  • __contains__ 完全独立于迭代状态,支持任意次数调用,且时间复杂度为 O(1)
  • __iter__ 和 __next__ 保持标准迭代器协议,与 for 循环、list() 等完全兼容;
  • 支持实例复用(如多次 in 查询 + 多次独立迭代)。

⚠️ 注意事项

  • 此方案牺牲了函数式生成器的简洁语法(yield),需手动管理状态(如 _current);
  • 若业务逻辑无法静态判定成员资格(例如需查数据库、解析流式文件),则 __contains__ 仍需遍历——此时应明确文档化该行为,或考虑缓存策略;
  • 切勿在 __contains__ 中修改迭代器内部状态(如重置 _current),否则会破坏迭代一致性。

总结:当“是否包含某值”可基于封闭规则(范围、公式、哈希映射等)提前判定时,用可迭代类替代生成器是唯一可靠、高效且符合 Python 协议的设计选择。它不是妥协,而是对抽象层级的精准把控——让 in 表达语义,让迭代表达过程。

热门AI工具

更多
DeepSeek
DeepSeek

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

豆包大模型
豆包大模型

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

通义千问
通义千问

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

腾讯元宝
腾讯元宝

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

文心一言
文心一言

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

讯飞写作
讯飞写作

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

即梦AI
即梦AI

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

ChatGPT
ChatGPT

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

相关专题

更多
class在c语言中的意思
class在c语言中的意思

在C语言中,"class" 是一个关键字,用于定义一个类。想了解更多class的相关内容,可以阅读本专题下面的文章。

581

2024.01.03

python中class的含义
python中class的含义

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

20

2025.12.06

数据库三范式
数据库三范式

数据库三范式是一种设计规范,用于规范化关系型数据库中的数据结构,它通过消除冗余数据、提高数据库性能和数据一致性,提供了一种有效的数据库设计方法。本专题提供数据库三范式相关的文章、下载和课程。

370

2023.06.29

如何删除数据库
如何删除数据库

删除数据库是指在MySQL中完全移除一个数据库及其所包含的所有数据和结构,作用包括:1、释放存储空间;2、确保数据的安全性;3、提高数据库的整体性能,加速查询和操作的执行速度。尽管删除数据库具有一些好处,但在执行任何删除操作之前,务必谨慎操作,并备份重要的数据。删除数据库将永久性地删除所有相关数据和结构,无法回滚。

2091

2023.08.14

vb怎么连接数据库
vb怎么连接数据库

在VB中,连接数据库通常使用ADO(ActiveX 数据对象)或 DAO(Data Access Objects)这两个技术来实现:1、引入ADO库;2、创建ADO连接对象;3、配置连接字符串;4、打开连接;5、执行SQL语句;6、处理查询结果;7、关闭连接即可。

355

2023.08.31

MySQL恢复数据库
MySQL恢复数据库

MySQL恢复数据库的方法有使用物理备份恢复、使用逻辑备份恢复、使用二进制日志恢复和使用数据库复制进行恢复等。本专题为大家提供MySQL数据库相关的文章、下载、课程内容,供大家免费下载体验。

259

2023.09.05

vb中怎么连接access数据库
vb中怎么连接access数据库

vb中连接access数据库的步骤包括引用必要的命名空间、创建连接字符串、创建连接对象、打开连接、执行SQL语句和关闭连接。本专题为大家提供连接access数据库相关的文章、下载、课程内容,供大家免费下载体验。

329

2023.10.09

数据库对象名无效怎么解决
数据库对象名无效怎么解决

数据库对象名无效解决办法:1、检查使用的对象名是否正确,确保没有拼写错误;2、检查数据库中是否已存在具有相同名称的对象,如果是,请更改对象名为一个不同的名称,然后重新创建;3、确保在连接数据库时使用了正确的用户名、密码和数据库名称;4、尝试重启数据库服务,然后再次尝试创建或使用对象;5、尝试更新驱动程序,然后再次尝试创建或使用对象。

417

2023.10.16

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

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

8

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号