Python列表组合计算技巧

本文深入解析了Python中itertools.permutations的实用要点与常见陷阱：它返回的是惰性迭代器而非直观列表，需显式转换或遍历才能查看结果；支持灵活指定排列长度r，兼顾全排列与部分排列需求；对重复元素不做自动去重，需结合set或专用工具谨慎处理；兼容字符串、元组等可迭代对象，但传入字典仅迭代键易引发误解；尤其提醒开发者——调试时直接print(permutations(...))只会看到内存地址，这是最常被忽略却极易导致“无输出”困惑的关键细节。

permutations 返回的是迭代器，不是列表

直接用 itertools.permutations 得到的结果是 itertools.permutations 对象，打印出来只显示类型和内存地址，不会自动展开。想看具体结果，必须手动转成列表或遍历。

• 错误写法：print(permutations([1,2,3])) → 输出类似
• 正确做法：list(permutations([1,2,3])) 或用 for p in permutations([1,2,3]): print(p)
• 注意：如果原列表很长（比如长度 ≥ 10），转成 list 可能瞬间吃光内存，优先考虑按需迭代

permutations 默认生成全排列，但可指定长度 r

permutations(iterable, r=None) 的第二个参数 r 控制每个排列的元素个数。不传时等价于 r=len(iterable)，即全排列；传了则只取 r 个元素的有序组合。

• list(permutations('ab', 2)) → [('a','b'), ('b','a')]
• list(permutations('abc', 2)) → [('a','b'), ('a','c'), ('b','a'), ('b','c'), ('c','a'), ('c','b')]
• 注意：r > len(iterable) 时返回空迭代器，不会报错，但 list(...) 得到空列表

输入含重复元素时，permutations 不去重

permutations 按位置计算，不关心值是否相同。若原始列表有重复值（如 [1,1,2]），结果中会出现语义重复的元组。

• list(permutations([1,1,2])) 返回 6 项，其中 (1,1,2) 和 (1,1,2) 看似一样，实则是不同位置的 1 组成的 —— 它们在内存中是两个独立对象，但值相等
• 若需真正去重，得套一层 set：list(set(permutations([1,1,2])))，但注意元组可哈希，而嵌套可变对象（如含列表）会报 TypeError
• 更稳妥的去重方式是先对输入排序 + 用 itertools.groupby，或改用 more-itertools.distinct_permutations（需额外安装）

字符串、元组、range 都能直接传入，但别传 dict

permutations 接收任意可迭代对象，常见如 str、tuple、range、list，但传 dict 会只迭代 key（Python 3.7+ 保持插入序），容易误以为在排 value。

• list(permutations('xyz')) → 字符串被当字符序列处理，结果是 ('x','y','z') 类型元组
• list(permutations(range(2))) → 等价于 permutations([0,1])
• list(permutations({1:'a', 2:'b'})) → 实际排的是 [1, 2]，不是 ['a','b']；要排 value，得显式写 permutations(list(d.values()))

实际用的时候，最常漏掉的是第一点：忘了它返回的是迭代器。调试时直接 print 看不到内容，就以为没结果，其实只是没触发迭代。

好了，本文到此结束，带大家了解了《Python列表组合计算技巧》，希望本文对你有所帮助！关注golang学习网公众号，给大家分享更多文章知识！