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标准库行为的一致性。这是开源社区协作改进代码质量的典型案例,展示了通过技术讨论和测试驱动开发可以实现的代码优化效果。
GLM-5智谱 AI 正式发布 GLM-5,旨在应对复杂系统工程和长时域智能体任务。Jinja00
GLM-5.1GLM-5.1是智谱迄今最智能的旗舰模型,也是目前全球最强的开源模型。GLM-5.1大大提高了代码能力,在完成长程任务方面提升尤为显著。和此前分钟级交互的模型不同,它能够在一次任务中独立、持续工作超过8小时,期间自主规划、执行、自我进化,最终交付完整的工程级成果。Jinja00
MiniMax-M2.7MiniMax-M2.7 是我们首个深度参与自身进化过程的模型。M2.7 具备构建复杂智能体应用框架的能力,能够借助智能体团队、复杂技能以及动态工具搜索,完成高度精细的生产力任务。Python00- QQwen3.5-397B-A17BQwen3.5 实现了重大飞跃,整合了多模态学习、架构效率、强化学习规模以及全球可访问性等方面的突破性进展,旨在为开发者和企业赋予前所未有的能力与效率。Jinja00
HY-Embodied-0.5这是一套专为现实世界具身智能打造的基础模型。该系列模型采用创新的混合Transformer(Mixture-of-Transformers, MoT) 架构,通过潜在令牌实现模态特异性计算,显著提升了细粒度感知能力。Jinja00
LongCat-AudioDiT-1BLongCat-AudioDiT 是一款基于扩散模型的文本转语音(TTS)模型,代表了当前该领域的最高水平(SOTA),它直接在波形潜空间中进行操作。00