Question

我有dict描述可能的配置值，例如：

{'a':[1,2], 'b':[3,4,5]}

我想生成所有可接受配置的列表，例如

[{'a':1, 'b':3},
 {'a':1, 'b':4},
 {'a':1, 'b':5},
 {'a':2, 'b':3},
 {'a':2, 'b':4},
 {'a':1, 'b':5}]

我查看了文档和SO，它似乎肯定涉及itertools.product，但我无法在没有嵌套for循环的情况下获得它。

Answer 1

您不需要在此处嵌套for循环：

from itertools import product
[dict(zip(d.keys(), combo)) for combo in product(*d.values())]

product(*d.values())生成所需的值组合，dict(zip(d.keys(), combo))重新组合每个组合。

演示：

>>> from itertools import product
>>> d = {'a':[1,2], 'b':[3,4,5]} 
>>> list(product(*d.values()))
[(1, 3), (1, 4), (1, 5), (2, 3), (2, 4), (2, 5)]
>>> [dict(zip(d.keys(), combo)) for combo in product(*d.values())]
[{'a': 1, 'b': 3}, {'a': 1, 'b': 4}, {'a': 1, 'b': 5}, {'a': 2, 'b': 3}, {'a': 2, 'b': 4}, {'a': 2, 'b': 5}]
>>> from pprint import pprint
>>> pprint(_)
[{'a': 1, 'b': 3},
 {'a': 1, 'b': 4},
 {'a': 1, 'b': 5},
 {'a': 2, 'b': 3},
 {'a': 2, 'b': 4},
 {'a': 2, 'b': 5}]

Answer 2

你也可以试试这个：

>>> dt={'a':[1,2], 'b':[3,4,5]} 
>>> [{'a':i,'b':j} for i in dt['a'] for j in dt['b']]
[{'a': 1, 'b': 3}, {'a': 1, 'b': 4}, {'a': 1, 'b': 5}, {'a': 2, 'b': 3}, {'a': 2, 'b': 4}, {'a': 2, 'b': 5}]

获取Python中所有可能的dict配置列表

2 个答案: