python - 在 Python 中生成所有可能映射的列表

Question

假设我们有一个映射 K -> V，其中定义域 K 是一个集合 ([1,2,3]) 并且共同定义域 V 从集合(['a', '公元前'])。是否有一种简洁的方法来枚举所有可能的映射(最好是字典列表或生成器):

例如。

[ { 1 : 'a', 2 : 'a', 3 : 'a' },
  { 1 : 'a', 2 : 'a', 3 : 'b' },
  { 1 : 'a', 2 : 'b', 3 : 'a' },
  ...
  { 1 : 'c', 2 : 'c', 3 : 'c' }
]

请注意，域的大小不固定，因此这种解决方案并不理想:

[ { 1 : x, 2 : y,  3 : z } for x in V for y in V for z in V ]

干杯

Answer 1

对itertools.product 使用repeat 参数:

K = set([1, 2, 3])
V = set(['a', 'b', 'c'])
itertools.product(V, repeat=len(K))

然后您可以在理解中构造 dict:

(dict(zip(K, x)) for x in itertools.product(V, repeat=len(K)))

检查:

>>> len([dict(zip([1, 2, 3], x)) for x in itertools.product('abc', repeat=3)])
27