1. 初识itertools.combinations()
第一次接触Python的itertools模块时,我就被combinations()这个函数深深吸引了。它就像是一个神奇的魔法盒,能够从一堆元素中变出所有可能的组合。想象你有一盒彩色积木,combinations()能帮你列出所有两两搭配的方案,而不用自己手动排列。
这个函数的基本语法非常简单:
from itertools import combinations
result = combinations(iterable, r)
其中iterable可以是任何可迭代对象(列表、字符串等),r则是你要选择的元素个数。比如我们有三个数字[1,2,3],想要找出所有两两组合:
nums = [1, 2, 3]
combs = list(combinations(nums, 2))
print(combs) # 输出:[(1, 2), (1, 3), (2, 3)]
这里有个关键点需要注意:combinations()返回的是迭代器对象,所以通常我们会用list()把它转换为列表。如果不转换,直接打印会看到类似<itertools.combinations object at 0x...>这样的输出,这表示它是一个待计算的生成器。
与permutations()函数不同,combinations()不考虑顺序。也就是说,(1,2)和(2,1)在排列中算两个结果,但在组合中只算一个。这个特性在实际应用中非常重要,特别是在需要去重的场景下。
2. 组合与排列的核心差异
很多初学者容易混淆组合(combinations)和排列(permutations)的概念。我刚开始学习时也经常搞混,直到用实际例子才彻底明白它们的区别。
假设我们有三张扑克牌:红桃A、方片K、梅花Q。现在要从中选两张:
- 组合(combinations)关心的是"有哪些不同的牌对",不考虑顺序。所以只有三种可能:(红桃A,方片K)、(红桃A,梅花Q)、(方片K,梅花Q)
- 排列(permutations)则关心"牌的顺序排列",所以会有六种可能,每种组合都有正反两种顺序
用代码来演示这个区别:
from itertools import combinations, permutations
cards = ['红桃A', '方片K', '梅花Q']
print("组合结果:", list(combinations(cards, 2)))
# 输出:[('红桃A', '方片K'), ('红桃A', '梅花Q'), ('方片K', '梅花Q')]
prin


1万+

被折叠的 条评论
为什么被折叠?



