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

使用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订单一致。

Answer 2

我认为通过反转字典键值对，您可以在以后方便地引用还原字典：

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']

由于原始字典中的值不是唯一的，因此您希望新字典的值为列表。

编辑：
为了使输出与产品保持一致：

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'])]

您可以使用相同的索引来获取键值对

识别产品的组合列表结果中每个元素的原始密钥

2 个答案: