一组值的所有唯一组合
如果我有任意数量的元素,每个元素都有一组指定的值,我该怎么做才能从每个元素的集合中获得一个值的所有可能组合?
例如,假设我有:
elems = {"A": (0, 1), "B": (-1, -5)}
我该怎么做才能获得以下内容?
({"A": 0, "B": -1}, {"A": 0, "B": -5}, {"A": 1, "B": -1}, {"A": 1, "B": -5})
输入/输出不需要使用 dicts,但我认为这是表示它的最简单方法。假设字典顺序很重要。
我想知道一种方法:
- 不假设示例中只有 2 个元素。
- 不假设所有元素的集合具有相等的长度。
这有点像列出真值表的所有输入。
回答
您可以使用嵌套列表/字典理解,itertools.product用于生成值的所有组合,然后将zip每个结果元组从该结果元组转换为键以生成键/值对以构成每个字典结果。这是一个使用稍微复杂的数据版本的示例:
import itertools
elems = {"A": (0, 1), "B": (-1, -5), "C": (1, 2, 3) }
keys = elems.keys()
res = [dict(zip(keys, p)) for p in itertools.product(*elems.values())]
输出:
[
{'A': 0, 'B': -1, 'C': 1},
{'A': 0, 'B': -1, 'C': 2},
{'A': 0, 'B': -1, 'C': 3},
{'A': 0, 'B': -5, 'C': 1},
{'A': 0, 'B': -5, 'C': 2},
{'A': 0, 'B': -5, 'C': 3},
{'A': 1, 'B': -1, 'C': 1},
{'A': 1, 'B': -1, 'C': 2},
{'A': 1, 'B': -1, 'C': 3},
{'A': 1, 'B': -5, 'C': 1},
{'A': 1, 'B': -5, 'C': 2},
{'A': 1, 'B': -5, 'C': 3}
]