more-itertools项目中is_sorted函数的优化与改进
more-itertools是一个提供额外迭代器工具的Python库,其中is_sorted函数用于判断一个可迭代对象是否已排序。最近社区对该函数的实现进行了深入讨论和优化,使其行为更加符合预期且性能更优。
原有实现的问题
原is_sorted函数实现存在几个关键问题:
-
逻辑双重否定:函数使用了与预期相反的比较操作符,然后通过not any来反转结果,这种双重否定增加了理解难度。
-
边界情况处理不一致:对于包含NaN值的列表,原函数会错误地返回True,而实际上NaN值会使排序变得无意义。
-
与内置sorted行为不一致:当可迭代对象包含特殊比较行为的元素时,函数结果与Python内置sorted函数的结果可能不同。
优化方案
经过社区讨论,最终采纳的优化方案具有以下特点:
-
直接使用符合直觉的比较操作:不再使用双重否定逻辑,而是直接使用与sorted函数一致的比较方式。
-
更高效的实现:通过itertools.tee创建两个迭代器,避免了starmap和pairwise的开销。
-
严格模式支持:通过strict参数支持严格排序检查,使用le(小于等于)或lt(小于)操作符。
优化后的核心代码如下:
def is_sorted(iterable, key=None, reverse=False, strict=False):
compare = le if strict else lt
it = iterable if key is None else map(key, iterable)
it1, it2 = tee(it)
next(it2 if reverse else it1, None)
return not any(map(compare, it1, it2))
性能对比
性能测试显示,新实现在大多数情况下都有显著提升:
- 对于100个元素的列表:6.3μs → 4.6μs (提升27%)
- 对于1000个元素的列表:59.3μs → 40.8μs (提升31%)
- 对于10000个元素的列表:583.9μs → 405.0μs (提升31%)
仅在极小列表(1个元素)时,新实现略慢(421.8ns → 629.9ns),这是由于初始化开销略高,但对实际应用影响很小。
技术考量
-
与sorted行为的一致性:新实现尽可能接近Python内置sorted函数的行为,使用相同的比较逻辑。
-
特殊值处理:正确处理NaN等特殊值,确保结果符合数学和编程直觉。
-
稳定性:通过全面测试确保在各种边界条件下都能正确工作,包括空列表、单元素列表、包含特殊对象的列表等。
总结
这次优化不仅提高了is_sorted函数的性能,更重要的是使其行为更加一致和可预测。通过简化比较逻辑和优化迭代器使用,函数现在能更好地服务于各种排序检查场景,同时保持与Python标准库行为的一致性。这是开源社区协作改进代码质量的典型案例,展示了通过技术讨论和测试驱动开发可以实现的代码优化效果。
atomcodeClaude Code 的开源替代方案。连接任意大模型,编辑代码,运行命令,自动验证 — 全自动执行。用 Rust 构建,极致性能。 | An open-source alternative to Claude Code. Connect any LLM, edit code, run commands, and verify changes — autonomously. Built in Rust for speed. Get StartedRust0152- DDeepSeek-V4-ProDeepSeek-V4-Pro(总参数 1.6 万亿,激活 49B)面向复杂推理和高级编程任务,在代码竞赛、数学推理、Agent 工作流等场景表现优异,性能接近国际前沿闭源模型。Python00
LongCat-Video-Avatar-1.5最新开源LongCat-Video-Avatar 1.5 版本,这是一款经过升级的开源框架,专注于音频驱动人物视频生成的极致实证优化与生产级就绪能力。该版本在 LongCat-Video 基础模型之上构建,可生成高度稳定的商用级虚拟人视频,支持音频-文本转视频(AT2V)、音频-文本-图像转视频(ATI2V)以及视频续播等原生任务,并能无缝兼容单流与多流音频输入。00
auto-devAutoDev 是一个 AI 驱动的辅助编程插件。AutoDev 支持一键生成测试、代码、提交信息等,还能够与您的需求管理系统(例如Jira、Trello、Github Issue 等)直接对接。 在IDE 中,您只需简单点击,AutoDev 会根据您的需求自动为您生成代码。Kotlin03
Intern-S2-PreviewIntern-S2-Preview,这是一款高效的350亿参数科学多模态基础模型。除了常规的参数与数据规模扩展外,Intern-S2-Preview探索了任务扩展:通过提升科学任务的难度、多样性与覆盖范围,进一步释放模型能力。Python00
skillhubopenJiuwen 生态的 Skill 托管与分发开源方案,支持自建与可选 ClawHub 兼容。Python0112