>>> from itertools import product, starmap
>>> d={'s1':{'a':[1,2,3],'b':[4,5,6]},'s2':{'d':[77,88,99],'e':[666,333,444]}}
>>> for k,v in sorted(d.items()): # use d.iteritems for py2
for x,y in zip(*starmap(product,sorted(v.items()))):
print k
print '{0},{1}'.format(*x)
print '{0},{1}'.format(*y)
print k
print
s1
a,1
b,4
s1
s1
a,2
b,5
s1
s1
a,3
b,6
s1
s2
d,77
e,666
s2
s2
d,88
e,333
s2
s2
d,99
e,444
s2
解释
它使用字典中每个列表的键获取值对,所以
{'a':[1,2,3],'b':[4,5,6]}
变成
[(('a', 1), ('b', 4)), (('a', 2), ('b', 5)), (('a', 3), ('b', 6))]
这是如何完成的:
第一行遍历 中的每个键和值d
。这些必须排序,所以我可以按升序遍历,因为字典没有顺序。
键的值是一个类似于下面的字典,它以键值对的元组排序。
d = {'a':[1,2,3],'b':[4,5,6]}
>>> sorted(d.items())
[('a', [1, 2, 3]), ('b', [4, 5, 6])]
然后product
用于获取与每个值配对的键。
>>> [product(*x) for x in sorted(d.items())]
# iterates through [('a', 1), ('a', 2), ('a', 3)], [('b', 4), ('b', 5), ('b', 6)]
这可以更简单地使用 来编写,尽管函数的参数(在这种情况下)来自元组starmap
,但它的构建方式类似于。见文档map
product
>>> starmap(product,sorted(d.items()))
# iterates through [('a', 1), ('a', 2), ('a', 3)], [('b', 4), ('b', 5), ('b', 6)]
然后列表是zippped
在一起的。
>>> zip(*starmap(product,sorted(d.items())))
[(('a', 1), ('b', 4)), (('a', 2), ('b', 5)), (('a', 3), ('b', 6))]