使用递归组合字典中的值



我有一个包含不同长度列表的字典,我想要一个通用算法来生成列表中所有元素的组合。下面给出了一段示例代码:

bob = {'a':['a','b','c'],     
   'b':[0],
   'c':['x','y']}

for i in bob['a']:
    for j in bob['b']:
        for k in bob['c']:
            print("%s - %s - %s"%(str(i),str(j),str(k)))

这是特定字典"bob"的所需输出,它生成输出:

a - 0 - x
a - 0 - y
b - 0 - x
b - 0 - y
c - 0 - x
c - 0 - y

但是,我想要一个通用算法。我可以使用递归来概括这一点,以便它可以处理具有任意数量的键的字典,如果是这样,如何?

额外信息:与键对应的值将始终是一维列表

一般算法itertools.product

>>> print(*itertools.product(*bob.values()), sep='n')
('a', 0, 'x')
('a', 0, 'y')
('b', 0, 'x')
('b', 0, 'y')
('c', 0, 'x')
('c', 0, 'y')

使用迭代工具中的产品,非常方便:

>>> from itertools import product
>>> for x in product(*[bob[k] for k in ('a', 'b', 'c')]):
...     print(' - '.join(map(str, x)))
...
a - 0 - x
a - 0 - y
b - 0 - x
b - 0 - y
c - 0 - x
c - 0 - y

如果 bob 键的顺序不相关,则可以将其简化为:

>>> for x in product(*bob.values()):
...     print(' - '.join(map(str, x)))

最新更新