将两个不同大小的列表组合成一个组合集

时间:2017-08-28 22:15:52

标签: python python-2.7 list

我试图将具有不同数据和大小的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中有没有一种干净的方法呢?

4 个答案:

答案 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))