python - 两个不同长度列表的循环组合

Question

我有两个清单：

list1 = ['a','b','c']
list2 = ['1','2','3','4','5']

我想列出清单：

list3 = [('1','a'),('2','b'),('3','c'),('4','a'),('5','b')]

换句话说，在它们之间做一个循环组合。所以，我的问题是：哪种方法更有效？

Answer 1

>>> from itertools import cycle
>>> list1 = ['a','b','c']
>>> list2 = ['1','2','3','4','5']
>>> zip(list2, cycle(list1))
[('1', 'a'), ('2', 'b'), ('3', 'c'), ('4', 'a'), ('5', 'b')]

正如一些人在评论中提到的，如果你想循环两个列表并获取前 n 个元素，你可以这样做，

>>> from itertools import islice, izip
>>> list(islice(izip(cycle(list2), cycle(list1)), 5))
[('1', 'a'), ('2', 'b'), ('3', 'c'), ('4', 'a'), ('5', 'b')]

Answer 2

这是另一种方法：永远继续循环组合的生成器：

def cc(l1, l2):
  i=0
  c1 = len(l1)
  c2 = len(l2)
  while True:
    yield (l1[i % c1], l2[i % c2])
    i += 1

你的问题的字面答案是：

x=cc(list2, list1)
[next(x) for i in range(max(len(list1), len(list2)))]
[('1', 'a'), ('2', 'b'), ('3', 'c'), ('4', 'a'), ('5', 'b')]

但是您现在有了一个灵活的基础，可以从中派生出各种其他有趣的部分。