Лучший способ перечислить декартово произведение с метками в python? - PullRequest
Новая
       4

Лучший способ перечислить декартово произведение с метками в python?

2 голосов
/ 29 ноября 2010

Учитывая словарь отображения переменных на возможные результаты: { '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 голосов
/ 29 ноября 2010 
[dict(zip(('lblA', 'lblB', 'lblC'), term)) for term in
  itertools.product((False, True) , (False, True), (0, 1, 2))]

EDIT:

Придирки, придирчивы ... 

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)]
...