Question

我有以下字典：

my_dict = {"A":0, "B":[1,2], "C":3}

我正在尝试将其拆分为包含每个可能值的2个词典的列表。例如：

#Desired result
[{"A":0, "B":1, "C":3}, {"A":0, "B":2, "C":3}]

我能够获得每个所需字典的列表，希望以后可以遍历该列表并使用update()合并字典。这将覆盖键"B"，并且仅创建一个字典。

这是我所做的：

dict_list = []
my_dict = {"A":0, "B":[1,2], "C":3}

for k, v in my_dict.items():
    if type(v) == list:
        for i in v:
            dict1 = {k:i}
            dict_list.append(dict1)
    if type(v) != list:
        dict2 = {k:v}
        dict_list.append(dict2)

new_dict = {}
for d in dict_list:
    new_dict.update(d)

print(new_dict)

输出：

{'A':0, 'B':2, 'C':3}

这将覆盖键'B'，并根据最后一个值仅创建一个字典。

Answer 1

如果您的字典只能有一个列表，这很容易。

>>> my_dict = {"A":0, "B":[1,2], "C":3}
>>> k, lst = next((k, v) for k, v in my_dict.items() if isinstance(v, list))
>>> [{**my_dict, k:x} for x in lst]
[{'A': 0, 'B': 1, 'C': 3}, {'A': 0, 'B': 2, 'C': 3}]

如果您的字典可以有多个列表，这会有些棘手。

>>> from itertools import product
>>> 
>>> my_dict = {"A":0, "B":[1,2], "C":[3, 4]}
>>> lists_kv = ([(k, x) for x in v] for k, v in my_dict.items() if isinstance(v, list))
>>> [{**my_dict, **dict(x)} for x in product(*lists_kv)]
[{'A': 0, 'B': 1, 'C': 3},
 {'A': 0, 'B': 1, 'C': 4},
 {'A': 0, 'B': 2, 'C': 3},
 {'A': 0, 'B': 2, 'C': 4}]

为字典中列表值的每种组合创建一个字典

1 个答案: