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标准库行为的一致性。这是开源社区协作改进代码质量的典型案例,展示了通过技术讨论和测试驱动开发可以实现的代码优化效果。
AutoGLM-Phone-9BAutoGLM-Phone-9B是基于AutoGLM构建的移动智能助手框架,依托多模态感知理解手机屏幕并执行自动化操作。Jinja00
Kimi-K2-ThinkingKimi K2 Thinking 是最新、性能最强的开源思维模型。从 Kimi K2 开始,我们将其打造为能够逐步推理并动态调用工具的思维智能体。通过显著提升多步推理深度,并在 200–300 次连续调用中保持稳定的工具使用能力,它在 Humanity's Last Exam (HLE)、BrowseComp 等基准测试中树立了新的技术标杆。同时,K2 Thinking 是原生 INT4 量化模型,具备 256k 上下文窗口,实现了推理延迟和 GPU 内存占用的无损降低。Python00
GLM-4.6V-FP8GLM-4.6V-FP8是GLM-V系列开源模型,支持128K上下文窗口,融合原生多模态函数调用能力,实现从视觉感知到执行的闭环。具备文档理解、图文生成、前端重构等功能,适用于云集群与本地部署,在同类参数规模中视觉理解性能领先。Jinja00
HunyuanOCRHunyuanOCR 是基于混元原生多模态架构打造的领先端到端 OCR 专家级视觉语言模型。它采用仅 10 亿参数的轻量化设计,在业界多项基准测试中取得了当前最佳性能。该模型不仅精通复杂多语言文档解析,还在文本检测与识别、开放域信息抽取、视频字幕提取及图片翻译等实际应用场景中表现卓越。00
GLM-ASR-Nano-2512GLM-ASR-Nano-2512 是一款稳健的开源语音识别模型,参数规模为 15 亿。该模型专为应对真实场景的复杂性而设计,在保持紧凑体量的同时,多项基准测试表现优于 OpenAI Whisper V3。Python00
GLM-TTSGLM-TTS 是一款基于大语言模型的高质量文本转语音(TTS)合成系统,支持零样本语音克隆和流式推理。该系统采用两阶段架构,结合了用于语音 token 生成的大语言模型(LLM)和用于波形合成的流匹配(Flow Matching)模型。 通过引入多奖励强化学习框架,GLM-TTS 显著提升了合成语音的表现力,相比传统 TTS 系统实现了更自然的情感控制。Python00
Spark-Formalizer-X1-7BSpark-Formalizer 是由科大讯飞团队开发的专用大型语言模型,专注于数学自动形式化任务。该模型擅长将自然语言数学问题转化为精确的 Lean4 形式化语句,在形式化语句生成方面达到了业界领先水平。Python00