列表字典的笛卡尔积

我正在尝试编写一些代码来测试一堆输入参数的笛卡尔积。

我看过itertools，但其product功能不完全是我想要的。有没有一种简单明显的方法来获取具有任意数量键的字典and每个值中的任意数量的元素，然后生成具有下一个排列的字典？

Input:

options = {"number": [1,2,3], "color": ["orange","blue"] }
print list( my_product(options) )

输出示例：

[ {"number": 1, "color": "orange"},
  {"number": 1, "color": "blue"},
  {"number": 2, "color": "orange"},
  {"number": 2, "color": "blue"},
  {"number": 3, "color": "orange"},
  {"number": 3, "color": "blue"}
]

好的，谢谢@dfan 告诉我我找错地方了。我现在已经得到了：

from itertools import product
def my_product(inp):
    return (dict(zip(inp.keys(), values)) for values in product(*inp.values())

EDIT：经过多年的Python经验，我认为更好的解决方案是接受kwargs而不是输入字典；调用风格更类似于原来的风格itertools.product。另外，我认为编写生成器函数，而不是返回生成器表达式的函数，可以使代码更清晰。所以：

import itertools
def product_dict(**kwargs):
    keys = kwargs.keys()
    for instance in itertools.product(*kwargs.values()):
        yield dict(zip(keys, instance))

如果你需要传递一个字典，list(product_dict(**mydict))。一个显着的变化是使用kwargs与任意输入类不同的是，它会阻止键/值排序，至少在 Python 3.6 之前是这样。