代码之家 › 专栏 › 技术社区 › user

用python枚举带有标签的笛卡尔产品的最佳方法?

itertools enumeration python

2

user · 技术社区 · 15 年前

给定一个字典,将变量映射到可能的结果: { 'lblA' : [False, True], 'lblB' : [False, True], 'lblC' : [0,1,2] }

我想列举所有可能的词典结果:

[ { 'lblA' : False , 'lblB' : False, 'lblC' : 0 },
{ 'lblA' : True , 'lblB' : False, 'lblC' : 0 },
{ 'lblA' : False , 'lblB' : True, 'lblC' : 0 },
{ 'lblA' : True , 'lblB' : True, 'lblC' : 0 },
{ 'lblA' : False , 'lblB' : False, 'lblC' : 1 },
{ 'lblA' : True , 'lblB' : False, 'lblC' : 1 },
{ 'lblA' : False , 'lblB' : True, 'lblC' : 1 },
{ 'lblA' : True , 'lblB' : True, 'lblC' : 1 },
{ 'lblA' : False , 'lblB' : False, 'lblC' : 2 },
{ 'lblA' : True , 'lblB' : False, 'lblC' : 2 },
{ 'lblA' : False , 'lblB' : True, 'lblC' : 2 },
{ 'lblA' : True , 'lblB' : True, 'lblC' : 2 } ]

我知道这可以递归完成,但我真的很想用 itertools 为了速度。

有人知道最好的方法吗?

非常感谢你的帮助!

编辑

我想为一本任意字典做这个。

1 回复 | 直到 15 年前

1

9

Ignacio Vazquez-Abrams 15 年前

[dict(zip(('lblA', 'lblB', 'lblC'), term)) for term in
  itertools.product((False, True) , (False, True), (0, 1, 2))]

编辑:

挑剔,挑剔。。。

src = {'lblA': (False, True), 'lblB': (False, True), 'lblC': (0, 1, 2)}

labels, terms = zip(*src.items())

print [dict(zip(labels, term)) for term in itertools.product(*terms)]