一组值的所有唯一组合

如果我有任意数量的元素,每个元素都有一组指定的值,我该怎么做才能从每个元素的集合中获得一个值的所有可能组合?

例如,假设我有:

elems = {"A": (0, 1), "B": (-1, -5)}

我该怎么做才能获得以下内容?

({"A": 0, "B": -1}, {"A": 0, "B": -5}, {"A": 1, "B": -1}, {"A": 1, "B": -5})

输入/输出不需要使用 dicts,但我认为这是表示它的最简单方法。假设字典顺序很重要。

我想知道一种方法:

  1. 不假设示例中只有 2 个元素。
  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}
]


以上是一组值的所有唯一组合的全部内容。
THE END
分享
二维码
< <上一篇
下一篇>>