讲师中心 微信公众号

工具推荐

视频效率加速
立刻MV
立刻MV

低成本歌曲 MV、视频生成

剧云
剧云

中文短剧剧本制作平台

Seko
Seko

AI短剧一键生成成片

AniShort
AniShort

AI短剧协同制作平台

Python itertools 常用迭代组合技巧

梦芳大大_6566

梦芳大大_6566

发布时间:2026-01-27 19:42:47

|

431人浏览过

|

来源于php中文网

原创

itertools.chain 是合并多个可迭代对象的最优解,惰性求值、内存友好;chain.from_iterable 适用于嵌套结构;groupby 需预排序才能正确分组;islice 是安全截取迭代器的唯一方式。

python itertools 常用迭代组合技巧

用 itertools.chain 合并多个可迭代对象,别再写 for 循环拼接

当你要把几个列表、生成器或文件行流串成一个连续流时,itertools.chain 是最轻量也最符合 Python 语义的解法。它不立即展开数据,保持惰性求值,内存友好。

  • 常见错误:用 + 拼接列表(触发复制,且只支持 list)或手动 for 循环 extend(破坏迭代器本质)
  • 正确写法:chain(list_a, list_b, range(3), my_generator()),返回一个迭代器,可直接用于 for 或传给 list()
  • 注意 chain.from_iterable 的适用场景:当你有一堆嵌套的可迭代对象(如 [[1,2], [3,4], (5,)]),它比 chain(*nested) 更安全(避免解包空序列报错)

itertools.groupby 必须先排序,否则分组结果不符合直觉

groupby 不是“按值归类”,而是“对连续相同键的元素分段”。如果输入没排好序,同一键的元素被分散,就会被拆成多组。

  • 典型翻车现场:对未排序的 ['a', 'b', 'a', 'c'] 用 groupby(x),得到三组:('a', ['a'])、('b', ['b'])、('a', ['a']) —— 第二个 'a' 不会和第一个合并
  • 正确做法:先用 sorted(data, key=key_func),再传给 groupby;若原始顺序不能丢,可考虑用 defaultdict(list) 替代
  • key 函数返回值必须可哈希;若需按对象属性分组,确保该属性稳定且可哈希(比如 lambda x: x.category)

用 itertools.islice 安全截取大迭代器,避开 list[10:20] 的陷阱

对生成器、文件对象或无限迭代器(如 count())做切片时,不能用方括号语法——那会触发 __getitem__,而多数迭代器不支持。必须用 islice。

Python Hardener
Python Hardener

通过完整的安全审查、代码整洁重构、健壮的错误处理、正确的日志记录以及完整的文档字符串覆盖,强化 Python 脚本。

下载
  • islice(iterator, start, stop, step) 返回新迭代器,不消耗原迭代器前面的项(除非 start > 0,此时会跳过前 start 项)
  • 注意:不能反向切片(stop 无效),也不能负索引;需要倒序取最后 N 项?得先转成 <code>deque(maxlen=N)
  • 常见误用:islice(my_gen(), 1000000, 1000010) 会跳过前一百万项——如果只是想取前 10 项,直接写 islice(gen, 10) 即可,别加多余参数

itertools.product 和 itertools.combinations_with_replacement 别混淆使用场景

这两个都生成组合,但语义完全不同,选错会导致逻辑错误或爆炸式增长。

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

  • product(A, B) 是笛卡尔积:所有 (a,b) 对,长度为 len(A) * len(B);常用于参数网格搜索、路径枚举
  • combinations_with_replacement(iterable, r) 是“可重复选 r 个的无序组合”:比如 combinations_with_replacement('AB', 2) → ('A','A'), ('A','B'), ('B','B');适合模拟抽样放回且不计顺序
  • 容易踩坑:permutations 和 combinations 默认不放回;若要放回且有序,只能用 product(比如密码暴力枚举:每个位置从字符集里独立选)

真正难的不是记住函数名,而是每次调用前确认:这个迭代器是否已耗尽?键是否已排序?切片范围是否超出生成能力?这些状态不会报错,只会静默返回空结果。

热门AI工具

更多
DeepSeek

DeepSeek是一款面向对话、写作、编程和推理场景的AI大模型工具。

UP简历
UP简历 Hot

一款AI办公效率工具,主要用于基于AI技术的免费在线简历制作工具,适合需要提升相关任务效率的用户。

二狗PPT
二狗PPT Hot

一款AI演示文稿工具,主要用于专为中式职场打造的AI PPT生成工具,适合需要提升相关任务效率的用户。

UpDream
UpDream Hot

一款AI视频创作工具,主要用于哔哩哔哩推出的自研AI视频创作工具,适合需要提升相关任务效率的用户。

豆包大模型

豆包大模型是一款由字节跳动推出的企业级大语言模型服务平台。

SkildArt
SkildArt Hot

SkildArt是一款AI文本写作工具,一站式 AI 视觉创作平台。

蛙蛙写作

一款AI论文写作工具,主要用于超级AI智能写作助手,适合需要提升相关任务效率的用户。

创客贴AI

创客贴AI是一款AI图像与设计工具,AI辅助的智能在线设计工具。

WorkBuddy

一款AI办公效率工具,主要用于腾讯云推出的AI原生桌面智能体工作台,适合需要提升相关任务效率的用户。

相关专题

更多
counta和count的区别
counta和count的区别

Count函数用于计算指定范围内数字的个数,而CountA函数用于计算指定范围内非空单元格的个数。本专题为大家提供相关的文章、下载、课程内容,供大家免费下载体验。

3828

2023.11.20

lambda表达式
lambda表达式

Lambda表达式是一种匿名函数的简洁表示方式,它可以在需要函数作为参数的地方使用,并提供了一种更简洁、更灵活的编码方式,其语法为“lambda 参数列表: 表达式”,参数列表是函数的参数,可以包含一个或多个参数,用逗号分隔,表达式是函数的执行体,用于定义函数的具体操作。本专题为大家提供lambda表达式相关的文章、下载、课程内容,供大家免费下载体验。

1151

2023.09.15

python lambda函数
python lambda函数

本专题整合了python lambda函数用法详解,阅读专题下面的文章了解更多详细内容。

345

2025.11.08

Python lambda详解
Python lambda详解

本专题整合了Python lambda函数相关教程,阅读下面的文章了解更多详细内容。

654

2026.01.05

C++ Lambda 表达式与函数式编程
C++ Lambda 表达式与函数式编程

深入讲解 C++ Lambda 表达式与函数式编程范式,涵盖 Lambda 基础语法与返回值推导、值捕获与引用捕获的区别与陷阱、初始化捕获(C++14 广义捕获)、泛型 Lambda(auto 参数)、mutable 关键字修改捕获变量、std::function 类型擦除与函数包装、std::bind 参数绑定(及其被 Lambda 替代的趋势)、Lambda 在 STL 算法(sort/transform/for_each/remo

829

2026.05.06

Java 函数式接口与 Lambda 进阶技巧
Java 函数式接口与 Lambda 进阶技巧

在 Java Stream 基础之上进一步深入函数式编程,涵盖 @FunctionalInterface 注解与自定义函数式接口设计、四种方法引用(静态/实例/对象/构造器)的使用场景、Lambda 中的 effectively final 变量捕获规则、函数组合(andThen/compose)与柯里化技巧、Comparator 链式比较器构建、CompletableFuture 异步函数式编排、Reactor / RxJava 响应

448

2026.05.08

堆和栈的区别
堆和栈的区别

堆和栈的区别:1、内存分配方式不同;2、大小不同;3、数据访问方式不同;4、数据的生命周期。本专题为大家提供堆和栈的区别的相关的文章、下载、课程内容,供大家免费下载体验。

6867

2023.07.18

堆和栈区别
堆和栈区别

堆(Heap)和栈(Stack)是计算机中两种常见的内存分配机制。它们在内存管理的方式、分配方式以及使用场景上有很大的区别。本文将详细介绍堆和栈的特点、区别以及各自的使用场景。php中文网给大家带来了相关的教程以及文章欢迎大家前来学习阅读。

2828

2023.08.10

Postman身份认证怎么配置
Postman身份认证怎么配置

本专题从实际API接口调试场景出发,讲解Postman身份认证的完整配置流程,包括常见认证协议介绍、授权参数设置、访问令牌填写、OAuth 2.0授权及认证失败排查,帮助用户正确完成接口授权与访问验证。

60

2026.11.06

热门下载

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

精品课程

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

Copyright 2014-2026 https://www.php.cn/ All Rights Reserved | php.cn