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