我试图将具有不同数据和大小的2个列表合并为1,并使用较小的列表" wrap"周围。我正在寻找一种干净的方法,例如。
输入:
list1 = ['apple', 'orange', 'strawberry', 'avocado']
list2 = ['1','2','3']
输出:
[
{"l1": "apple", "l2": "1"},
{"l1": "orange", "l2": "2"},
{"l1": "strawberry", "l2": "3"},
{"l1": "avocado", "l2": "1"}
]
请注意,对于"avocado"
,我们回到了"1"
并将其包裹在list2中。
明显(丑陋)的解决方案是从一个空列表开始,在一个循环中有两个索引,每个迭代附加一个新的列表项,而较小的一个包装'到达结束时的开始。在Python 2.7中有没有一种干净的方法呢?
答案 0 :(得分:6)
您可以使用itertools.cycle
来包装第二个列表:
from itertools import cycle
lst = [dict(zip(['l1', 'l2'], tup)) for tup in zip(list1, cycle(list2))]
答案 1 :(得分:2)
您可以使用避免附加到空列表的生成器:
def func(l1, l2):
length1 = len(l1)
length2 = len(l2)
for idx in range(max(length1, length2)):
# I use the modulo so the indices wrap around.
yield {'l1': l1[idx % length1], 'l2': l2[idx % length2]}
list(func(list1, list2))
# [{'l1': 'apple', 'l2': '1'},
# {'l1': 'orange', 'l2': '2'},
# {'l1': 'strawberry', 'l2': '3'},
# {'l1': 'avocado', 'l2': '1'}]
然而itertools.cycle
(见另一个答案)可能要好得多。
答案 2 :(得分:2)
为简单起见,您只能使用enumerate
;但是,更清洁的解决方案将涉及itertools.cycle
:
list1 = ['apple', 'orange', 'strawberry', 'avocado']
list2 = ['1','2','3']
new_list = [{"li":a, "l2":list2[i%len(list2)]} for i, a in enumerate(list1)]
输出:
[{'l2': '1', 'li': 'apple'}, {'l2': '2', 'li': 'orange'}, {'l2': '3', 'li': 'strawberry'}, {'l2': '1', 'li': 'avocado'}]
答案 3 :(得分:0)
def processing(opt, longlist, shortlist):
def processing_iter(inL, refL, outL):
if refL == []:
return outL
elif inL == []:
return processing_iter(shortlist, refL[1:], outL+opt(refL[0], shortlist[0]))
else:
return processing_iter(inL[1:], refL[1:], outL+opt(refL[0], inL[0]))
return processing_iter(shortlist, longlist, [])
def makedict(a,b): return [{"l1":a, "l2":b}]
list1 = ['apple', 'orange', 'strawberry', 'avocado']
list2 = ['1','2','3']
print(processing(makedict,list1, list2))