在Python中如何枚举带标签的笛卡尔积?

2 投票
1 回答
1709 浏览
提问于 2025-04-16 07:44

给定一个字典,它把变量和可能的结果对应起来:

{ '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 个回答

9
[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)]

编辑:

挑剔,挑剔……

撰写回答