Python создает комбинацию всех словарей на основе списков значений

#python #python-3.x #dictionary #combinations #itertools

#python #python-3.x #словарь #комбинации #itertools

Вопрос:

Я хотел бы создать словари из определенного количества диапазонов переменных, как показано ниже.

У меня есть действительно базовое рабочее решение, но мне было интересно, есть ли какой-нибудь более лаконичный способ выразить ту же идею (я чувствую, что это можно решить с помощью 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)
 

Спасибо

Комментарии:

1. Я просто скопировал вставленный выше код и выполнил, и он выводит все комбинации на консоль.

Ответ №1:

Возьмите декартово произведение:

 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 конструктор для создания словаря из итератора пар ключ / значение.