python - Itertools 组合无重复 : Where rgb is equivelant to rbg etc

Question

我正在尝试使用 itertools.combinations 返回唯一的组合。我搜索了几个类似的问题，但未能找到答案。

一个例子:

>>> import itertools
>>> e = ['r','g','b','g']
>>> list(itertools.combinations(e,3))
[('r', 'g', 'b'), ('r', 'g', 'g'), ('r', 'b', 'g'), ('g', 'b', 'g')]

就我的目的而言，(r,g,b) 与 (r,b,g) 相同，因此我只想返回 (rgb)、(rgg) 和 (gbg)。

这只是一个说明性示例，我想忽略所有此类“重复项”。列表 e 最多可以包含 5 个元素。每个单独的元素可以是 r、g 或 b。始终从 e 中寻找 3 个元素的组合。

具体而言，以下是我希望称为“有效”的唯一组合:(rrr)、(ggg)、(bbb)、(rgb)。

因此，也许问题可以归结为如何将 (rgb) 的任何变体视为等于 (rgb)，从而忽略它。

我可以使用itertools来实现这一点还是我需要编写自己的代码来删除此处的“重复项”？如果没有 itertools 解决方案，那么我可以轻松检查每个是否都是 (rgb) 的变体，但这感觉有点“un-pythonic”。

Answer 1

您可以使用set来丢弃重复项。

在您的情况下，字符数是识别重复项的方式，因此您可以使用collections.Counter。为了将它们保存在 set 中，您需要将它们转换为 frozenset(因为 Counter 不可散列):

>>> import itertools
>>> from collections import Counter
>>> e = ['r','g','b','g']
>>> result = []
>>> seen = set()
>>> for comb in itertools.combinations(e,3):
...     cnts = frozenset(Counter(comb).items())
...     if cnts in seen:
...         pass
...     else:
...         seen.add(cnts)
...         result.append(comb)
>>> result
[('r', 'g', 'b'), ('r', 'g', 'g'), ('g', 'b', 'g')]

如果要将它们转换为字符串，请使用:

result.append(''.join(comb))  # instead of result.append(comb)

它会给出:

['rgb', 'rgg', 'gbg']

该方法是 unique_everseen recipe (itertools module documentation) 的变体- 所以它可能“相当Pythonic”。