Python根据值列表创建所有字典的组合
我想从一定数量的变量范围创建字典,如下所示。
我确实有一个非常基本的工作解决方案,但我想知道是否有一些更简洁的方式来表达相同的想法(我觉得它可以用 itertools 解决?)
a = [1, 2, 3]
b = ["me", "you", "her"]
c = range(1, 5)
d = ["mon", "tues", "wed"]
dic = {"a": None, "b": None, "c": None, "d": None}
for i in a:
dic["a"] = i
for j in b:
dic["b"] = j
for k in c:
dic["c"] = k
for l in d:
dic["d"] = l
print(dic)
谢谢
回答
以笛卡尔积为例:
import itertools
a = [1, 2, 3]
b = ["me", "you", "her"]
c = range(1, 5)
d = ["mon", "tues", "wed"]
names = ["a", "b", "c", "d"]
vals = [a, b, c, d]
for comb in itertools.product(*vals):
dic = dict(zip(names, comb))
print(dic)
这用于zip获取元组的迭代器,其中第一个元素是名称,第二个元素是值,然后使用dict构造函数从键/值对的迭代器中创建字典。