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