Question

我有一些组数据，如下面的a，b，c等。我想要做的是从数据中计算所有可能的组合。如果输入数据的计数总是相同的，那就没问题。但在我的情况下，我想假设输入数据的计数从0到N，我的意思是a，b，c，d，e ......

我想我必须使用循环循环。但我不确定使用循环循环来解决这个问题。

输入

a = ["x", "y"]
b = ["q", "w", "c"]
c = ["i", "o", "p"]
...

输出继电器

预期输出是与每个值的所有组合的所有组合。

[{a:[], b:["q"], c:["i", "o"]}, {a:["x"], b:[], c:["o"]}, ...]

Answer 1

如果我了解您正在寻找的内容，则可以使用itertools.product()而不是powersets输入（文档中有powerset()的配方）。 map()函数可用于将 powerset 应用于每个输入。

from itertools import product, combinations, chain
from pprint import pprint

def powerset(iterable):
    "powerset([1,2,3]) --> () (1,) (2,) (3,) (1,2) (1,3) (2,3) (1,2,3)"
    s = list(iterable)
    return chain.from_iterable(combinations(s, r) for r in range(len(s)+1))

names = ['a', 'b', 'c']
a = ["x", "y"]
b = ["q", "w", "c"]
c = ["i", "o", "p"]

result = []
for values in product(*map(powerset, [a, b, c])):
    result.append(dict(zip(names, values)))
pprint(result)

以下是它的工作原理：

首先，它构建了powersets：

>>> list(powerset(["x", "y"]))
[(), ('x',), ('y',), ('x', 'y')]
>>> 
>>> list(powerset(["x", "y"]))
[(), ('x',), ('y',), ('x', 'y')]
>>> list(powerset(["q", "w", "c"]))
[(), ('q',), ('w',), ('c',), ('q', 'w'), ('q', 'c'),
 ('w', 'c'), ('q', 'w', 'c')]
>>> list(powerset(["i", "o", "p"]))
[(), ('i',), ('o',), ('p',), ('i', 'o'), ('i', 'p'), 
 ('o', 'p'), ('i', 'o', 'p')]

下一个product()从每个powerset中提取一个元素：

>>> for values in product(*map(powerset, [a, b, c])):
        print(values)

((), (), ())
((), (), ('i',))
((), (), ('o',))
((), (), ('p',))
((), (), ('i', 'o'))
((), (), ('i', 'p'))
((), (), ('o', 'p'))
((), (), ('i', 'o', 'p'))
((), ('q',), ())
((), ('q',), ('i',))
((), ('q',), ('o',))
((), ('q',), ('p',))
((), ('q',), ('i', 'o'))
((), ('q',), ('i', 'p'))
((), ('q',), ('o', 'p'))
((), ('q',), ('i', 'o', 'p'))

最后，我们zip()将上述结果与变量 names 结合起来制作dict()：

# What zip does
>>> list(zip(['a', 'b', 'c'], ((), ('q',), ('i', 'o', 'p'))))
[('a', ()), ('b', ('q',)), ('c', ('i', 'o', 'p'))]

# What dict does with the zip:
>>> dict(zip(['a', 'b', 'c'], ((), ('q',), ('i', 'o', 'p'))))
{'b': ('q',), 'c': ('i', 'o', 'p'), 'a': ()}

在Python中计算来自不同数组的所有可能组合

1 个答案: