Определение оригинального ключа каждого элемента в комбинированном списке результатов продукта

Question

У меня есть словарь, подобный этому:

d = {'o1': [1, 2, 3], 'o2': [0, -1], 'o3': [7, 8, 10, 11]}

И я хочу, чтобы все комбинации каждого элемента в каждом списке. Для этого я использую продукт:

itertools.product(*d.values())

Пройдя каждую комбинацию, вы получите:

(7, 0, 1)
(7, 0, 2)
(7, 0, 3)

Мне нужно отслеживать исходный список каждого элемента в результирующем списке комбинаций. Мне нужно иметь возможность сказать, что 7 принадлежит 'o3', 0 к 'o2' и 1 к 'o1'. Элементы не уникальны.

Как я мог пойти об этом?

Я пробовал это:

r = [[(k, x) for x in v] for k, v in d.iteritems()]
itertools.product(*r)

Это дает:

(('o3', 7), ('o2', 0), ('o1', 1))
(('o3', 7), ('o2', 0), ('o1', 2))
(('o3', 7), ('o2', 0), ('o1', 3))

Но я не думаю, что это слишком элегантно.

Answer 1

Я думаю, что, инвертировав пару значений ключа словаря, вы можете позже обратиться к возвращенному словарю удобно:

dd = dict()

for key, val in d.items():
  for element in val:
    if not dd.get(element):
      dd[element] = list()
    dd[element].append(key)

dd
# {1: ['o1'], 2: ['o1'], 3: ['o1'], 0: ['o2'], -1: ['o2'], 7: ['o3'], 8: ['o3'], 10: ['o3'], 11: ['o3']}

dd[1]
# ['o1']

Поскольку значения в исходном словаре не являются уникальными, вы хотите, чтобы значения нового словаря были списками.

Edit:
Для того, чтобы выровнять вывод с продуктом:

p = list(product (*d.values())
[tuple(dd[i] for i in val) for val in p]
# [(['o1'], ['o2'], ['o3']), (['o1'], ['o2'], ['o3']), (['o1'], ['o2'], ['o3']), (['o1'], ['o2'], ['o3']), (['o1'], ['o2'], ['o3']), (['o1'], ['o2'], ['o3']), (['o1'], ['o2'], ['o3']), (['o1'], ['o2'], ['o3']), (['o1'], ['o2'], ['o3']), (['o1'], ['o2'], ['o3']), (['o1'], ['o2'], ['o3']), (['o1'], ['o2'], ['o3']), (['o1'], ['o2'], ['o3']), (['o1'], ['o2'], ['o3']), (['o1'], ['o2'], ['o3']), (['o1'], ['o2'], ['o3']), (['o1'], ['o2'], ['o3']), (['o1'], ['o2'], ['o3']), (['o1'], ['o2'], ['o3']), (['o1'], ['o2'], ['o3']), (['o1'], ['o2'], ['o3']), (['o1'], ['o2'], ['o3']), (['o1'], ['o2'], ['o3']), (['o1'], ['o2'], ['o3'])]

Вы можете использовать ту же индексацию, чтобы получить пару ключ-значение

Answer 2

Используйте collections.OrderedDict, чтобы указать порядок ваших значений.

Ниже приведены несколько способов сделать это.

from collections import OrderedDict

od1 = OrderedDict(sorted(d.items()))

od2 = OrderedDict([('o1', [1, 2, 3]), ('o2', [0, -1]), ('o3', [7, 8, 10, 11])])

Затем, когда вы запустите itertools.product, порядок итераций будет соответствовать вашему OrderedDict порядку.