首页 > 解决方案 > python中两个列表的乘积

问题描述

如何组合 2 个列表,使列表 2 中的每个元素与列表 1 中的元素组合,以便“a”在每次迭代中仅与列表 1 中的一个元素组合,然后“b”仅与列表中的两个元素组合在每次迭代中列出 1。

list1= [1,2,3]

list2= [‘a’,‘b’,‘c’,‘d’,‘e’]

dict_list_2 = {‘a’ : 1 , ‘b’ : 2 , ‘c’ : 3 , ‘d’ : 4 , ‘e’ :5}

Expected Output: ('a',1) (a ,2 ) (a , 3) (b , 1 ,2) (b , 1 ,3) (b , 2 ,3) (b , 1 ,1) (b , 2 ,2) (b , 3 ,3) – 

在此处输入图像描述

标签: pythonlistdictionaryitertools

解决方案


假设您想要所有带有替换的组合list1

import itertools

list1= [1,2,3]
list2= list("abcde")
dict_list_2 = {"a" : 1 , "b" : 2 , "c" : 3 , "d" : 4 , "e" :5}

[(letter,) + product
 for letter in list2
 for product in itertools.product(list1, repeat=dict_list_2[letter])]

# out:
[('a', 1),
 ('a', 2),
 ('a', 3),
 ('b', 1, 1),
 ('b', 1, 2),
 ('b', 1, 3),
 ('b', 2, 1),
 ('b', 2, 2),
 ('b', 2, 3),
 ('b', 3, 1),
 ('b', 3, 2),
 ('b', 3, 3),
 ('c', 1, 1, 1),
 ('c', 1, 1, 2),
 ('c', 1, 1, 3),
 ('c', 1, 2, 1),
...

推荐阅读