如何追踪itertools.product组合结果中各元素的原始字典键

这问题很常见，核心就是要让组合里的每个元素和它对应的原字典键绑定起来，其实只要保证键的顺序和传给itertools.product的列表顺序一致，就能轻松实现。

具体步骤和代码示例 #

首先，我们需要先拿到和d.values()顺序完全对应的键列表，然后遍历组合时用zip()把键和元素配对：

import itertools

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

# 获取与d.values()顺序匹配的键列表（Python3.7+字典会保留插入顺序）
source_keys = list(d.keys())

# 遍历所有组合，同时关联元素的来源键
for combo in itertools.product(*d.values()):
    # 生成键-元素对应的字典，方便直接查看每个元素的来源
    combo_with_sources = dict(zip(source_keys, combo))
    print(combo_with_sources)

运行这段代码，你会得到类似这样的输出：

{'o1': 1, 'o2': 0, 'o3': 7}
{'o1': 1, 'o2': 0, 'o3': 8}
{'o1': 1, 'o2': 0, 'o3': 10}
...

针对你例子中的特殊情况 #

你提到会得到(7, 0, 1)这样的组合，这说明你传给itertools.product的列表顺序是o3、o2、o1对应的列表（也就是颠倒了原字典的values顺序）。这种情况下，只需要同步颠倒键的顺序即可：

# 如果product的输入是颠倒后的values
reversed_values = reversed(d.values())
reversed_keys = list(reversed(d.keys()))

for combo in itertools.product(*reversed_values):
    combo_with_sources = dict(zip(reversed_keys, combo))
    print(combo_with_sources)  # 会得到{'o3':7, 'o2':0, 'o1':1}这样的结果

关键注意点 #

• Python版本兼容性：Python3.7及以上的普通字典会保留插入顺序，所以list(d.keys())和d.values()的顺序是完全对应的。如果是Python3.6及更早版本，需要用collections.OrderedDict来定义你的字典，确保键值顺序一致。
• 灵活的输出形式：如果你不想用字典，也可以用list(zip(source_keys, combo))得到类似[('o1',1), ('o2',0), ('o3',7)]的元组列表，方便后续处理。

内容的提问来源于stack exchange，提问作者dabadaba