如何在Python中从dict of list中的值生成所有组合

Question

我想生成dict中索引列表中的所有值组合：

{'A':['D','E'],'B':['F','G','H'],'C':['I','J']}

每次，每个判决条目的一个项目将被选择并组合到其他键的项目中，所以我有：

['D','F','I']
['D','F','J']
['D','G','I']
['D','G','J']
['D','H','I']
...
['E','H','J']

我知道在itertools中可以生成列表中项目的组合，但我不认为我可以在这里使用它，因为我有不同的值“池”。

是否有任何现有的解决方案可以做到这一点，或者我应该如何着手自己做到这一点，我完全被这种嵌套结构所困扰。

Answer 1

如何使用的参数网格从Scikit学习？它创建了一个生成器，您可以在正常的for循环中迭代它。在每次迭代中，您将拥有一个包含当前参数组合的字典。

from sklearn.model_selection import ParameterGrid

params = {'A':['D','E'],'B':['F','G','H'],'C':['I','J']}
param_grid = ParameterGrid(params)
for dict_ in param_grid:
    # Do something with the current parameter combination in ``dict_``
    print(dict_["A"])
    print(dict_["B"])
    print(dict_["C"])

Answer 2

如果你想保持key： value在排列中，你可以使用：

import itertools
keys, values = zip(*my_dict.items())
permutations_dicts = [dict(zip(keys, v)) for v in itertools.product(*values)]

这将为您提供一个包含排列的目录列表：

print(permutations_dicts)
[{'A':'D', 'B':'F', 'C':'I'}, 
 {'A':'D', 'B':'F', 'C':'J'},
 ...
 ]

免责声明不完全是OP的要求，但谷歌让我来这里寻找。

Answer 3

import itertools as it

my_dict={'A':['D','E'],'B':['F','G','H'],'C':['I','J']}
allNames = sorted(my_dict)
combinations = it.product(*(my_dict[Name] for Name in allNames))
print(list(combinations))

其中打印：

[('D', 'F', 'I'), ('D', 'F', 'J'), ('D', 'G', 'I'), ('D', 'G', 'J'), ('D', 'H', 'I'), ('D', 'H', 'J'), ('E', 'F', 'I'), ('E', 'F', 'J'), ('E', 'G', 'I'), ('E', 'G', 'J'), ('E', 'H', 'I'), ('E', 'H', 'J')]