获取每个项目的字典,并在一行中获取新列表

时间:2019-05-28 07:24:30

标签: python

我有一个字典,像这样:

{'brand': ['Three Squirrels', 'liang', 'BE'],
  'op': ['交', '差'],
  'cate': ['aaaa','bbbb','cccc']
}

现在,我要创建一个新列表,像这样:

 [
 {'brand':'Three Squirrels','op':'交','cate':'aaaa'},
 {'brand':'Three Squirrels','op':'差','cate':'bbbb'},
 {'brand':'liang','op':'交','cate':'aaaa'},
 {'brand':'liang','op':'交','cate':'bbbb'},
 .....
 .....
]

列表len为3 * 2 * 3 = 18

如何在一行代码中做到这一点?

1 个答案:

答案 0 :(得分:3)

您可以通过itertools.product生成所有可能的值 然后,您可以遍历值并通过将键和值压缩在一起来创建字典列表

from itertools import product

dct = {'brand': ['Three Squirrels', 'liang', 'BE'],
  'op': ['交', '差'],
  'cate': ['aaaa','bbbb','cccc']
}

#Get the list of keys
keys = dct.keys()

#Iterate through the values, and zip key and value together
result = [ dict(zip(keys, item)) for item in product(*dct.values())]

print(result)
print(len(result))

输出将为

[{'brand': 'Three Squirrels', 'op': '交', 'cate': 'aaaa'}, 
{'brand': 'Three Squirrels', 'op': '交', 'cate': 'bbbb'}, 
{'brand': 'Three Squirrels', 'op': '交', 'cate': 'cccc'}, 
{'brand': 'Three Squirrels', 'op': '差', 'cate': 'aaaa'},
 {'brand': 'Three Squirrels', 'op': '差', 'cate': 'bbbb'}, 
{'brand': 'Three Squirrels', 'op': '差', 'cate': 'cccc'}, 
{'brand': 'liang', 'op': '交', 'cate': 'aaaa'}, 
{'brand': 'liang', 'op': '交', 'cate': 'bbbb'}, 
{'brand': 'liang', 'op': '交', 'cate': 'cccc'}, 
{'brand': 'liang', 'op': '差', 'cate': 'aaaa'}, 
{'brand': 'liang', 'op': '差', 'cate': 'bbbb'}, 
{'brand': 'liang', 'op': '差', 'cate': 'cccc'}, 
{'brand': 'BE', 'op': '交', 'cate': 'aaaa'}, 
{'brand': 'BE', 'op': '交', 'cate': 'bbbb'}, 
{'brand': 'BE', 'op': '交', 'cate': 'cccc'}, 
{'brand': 'BE', 'op': '差', 'cate': 'aaaa'}, 
{'brand': 'BE', 'op': '差', 'cate': 'bbbb'}, 
{'brand': 'BE', 'op': '差', 'cate': 'cccc'}]

18