使用itertools限制排列

如何解决使用itertools限制排列

我正在编写一个工具，该工具希望从长度为13个字符的列表中生成所有可能的配对对。每对必须交换第二个元素。即要从列表ABCDEFGHIJKLM中生成诸如AB CD，AB CE或CD GK之类的对。

我正在使用itertools.permutations来做到这一点，并且有效：

perm_pairs = itertools.permutations(my_list,4)

我打算遍历perm_pairs并在单独的程序中使用这些对。但是，由于顺序不重要，并且后续过程具有较高的时间复杂度，因此我想通过防止配对反转来提高效率。也就是说，如果我已经拥有AB CD，我不想生成CD AB。

itertools.combinations()不会产生所有必需的对，但是itertools.permeations()生成的对比迭代所需的对多4倍。

是否有避免发生反转的“中间方法”？

解决方法

您可以为每个字母对使用itertools.permutations，然后检查某对（perm_pairs）中对的对数是否不存在：

from itertools import permutations

my_list = "ABCDEFGHIJKLM"

perm_pairs = set()

for pair_1 in permutations(my_list,2):
  for pair_2 in permutations((c for c in my_list if c not in pair_1),2):
    if pair_2 + pair_1 not in perm_pairs:
      perm_pairs.add(pair_1 + pair_2)