在 Python 中测试所有组合

Question

我有两组选项：

optionList1 = [a1,a2,a3,...,an]
optionList2 = [b1,b2,b3,...,bn]

选项列表中的元素数量不一定相等，我必须从第一个选项列表中选择两次。如何确保我已尝试了第一个列表中的 2 个选项和第二个列表中的一个选项的每种组合。下面是一个选择集示例...

selectedOptions = [an1,an2,bn]

Answer 1

假设您不希望第一个列表中出现重复项，则合并来自 itertools 的 product 和 permutations：

>>> from itertools import product,permutations
>>> o1 = 'a1 a2 a3'.split()
>>> o2 = 'b1 b2 b3'.split()
>>> for (a,b),c in product(permutations(o1,2),o2):
...     print a,b,c
... 
a1 a2 b1
a1 a2 b2
a1 a2 b3
a1 a3 b1
a1 a3 b2
a1 a3 b3
a2 a1 b1
a2 a1 b2
a2 a1 b3
a2 a3 b1
a2 a3 b2
a2 a3 b3
a3 a1 b1
a3 a1 b2
a3 a1 b3
a3 a2 b1
a3 a2 b2
a3 a2 b3

Answer 2

假设您不希望 list1 中出现重复的条目，则可以使用以下生成器来迭代所有组合：

def combinations(list1, list2):
    return ([opt1, opt2, opt3]
            for i,opt1 in enumerate(list1)
            for opt2 in list1[i+1:]
            for opt3 in list2)

但是，这不会以不同的顺序从 list1 中选择相同的选项。如果你想同时获得 [a1, a2, b1] 和 [a2, a1, b1] 那么你可以使用：

def combinations(list1, list2):
    return ([opt1, opt2, opt3]
            for opt1 in list1
            for opt2 in list1
            for opt3 in list2 if opt1 != opt2)

Answer 3

您可以使用 itertools.product 来实现此目的。它返回所有可能的组合。

例如，

for a1, a2, b in itertools.product(optionlist1,optionlist1,optionlist2):
    do_something(a1,a2,b)

这将产生“双精度”为 [a1,a1,b2] 和 [a2,a3,b2],[a3,a2,b2]。您可以使用过滤器解决此问题。以下内容可防止任何双打*：

for a1,a2,b in itertools.ifilter(lambda x: x[0]<x[1], itertools.product(optionlist1,optionlist1,optionlist2)):
    do_something(a1,a2,b)

(*) 这假设选项具有某种自然顺序，所有原始值都是这种情况。

尚的答案也很好。我写了一些代码来比较它们：

from itertools import ifilter, product
import random
from timeit import repeat

def generator_way(list1, list2):
    def combinations(list1, list2):
        return ([opt1, opt2, opt3]
                for i,opt1 in enumerate(list1)
                for opt2 in list1[i+1:]
                for opt3 in list2)
    count = 0
    for a1,a2,b in combinations(list1,list2):
        count += 1

    return count

def itertools_way(list1,list2):
    count = 0
    for a1,a2,b in ifilter(lambda x: x[0] < x[1], product(list1,list1,list2)):
        count += 1
    return count

list1 = range(0,100)
random.shuffle(list1)
list2 = range(0,100)
random.shuffle(list2)

print sum(repeat(lambda: generator_way(list1,list2),repeat = 10, number=1))/10
print sum(repeat(lambda: itertools_way(list1,list2),repeat = 10, number=1))/10

结果是：

0.189330005646
0.428138256073

所以生成器方法更快。然而，速度并不是一切。就我个人而言，我发现我的代码“更干净”，但选择权在你！

（顺便说一句，它们给出了相同的计数，因此两者同样正确。）

Answer 4

在我看来，您正在寻找 itertools.product()

>>> options_a = [1,2]
>>> options_b = ['a','b','c']
>>> list(itertools.product(options_a, options_a, options_b))
[(1, 1, 'a'),
 (1, 1, 'b'),
 (1, 1, 'c'),
 (1, 2, 'a'),
 (1, 2, 'b'),
 (1, 2, 'c'),
 (2, 1, 'a'),
 (2, 1, 'b'),
 (2, 1, 'c'),
 (2, 2, 'a'),
 (2, 2, 'b'),
 (2, 2, 'c')]

Answer 5

一种方法是使用 itertools.product。

for x, y, z in itertools.product(optionlist1,optionlist1,optionlist2):
    print x,y,z