Python如何按特定规则合并两个列表?遍历a并循环b生成新列表求助
列表循环合并的实现方案
给定两个列表:
a = ['a', 'b', 'c', 'd', 'e'] b = [1, 2, 3]
需要生成结果列表:['a1', 'b2', 'c3', 'd1', 'e2'],即遍历a的同时循环复用b的元素进行拼接。
你之前使用zip(a,b)的写法只能得到前3个元素,因为zip会在两个列表中较短的那个耗尽时停止迭代,无法覆盖a的全部元素。
方法一:使用itertools.cycle循环复用b
借助itertools模块的cycle函数,可以创建一个无限循环迭代b元素的迭代器,再和a配对拼接:
from itertools import cycle result = [f"{char}{num}" for char, num in zip(a, cycle(b))] print(result) # 输出:['a1', 'b2', 'c3', 'd1', 'e2']
方法二:手动索引取模(无需额外模块)
通过计算索引对b的长度取模,实现循环获取b的元素:
result = [f"{a[i]}{b[i % len(b)]}" for i in range(len(a))] print(result) # 输出:['a1', 'b2', 'c3', 'd1', 'e2']
内容的提问来源于stack exchange,提问作者Rohith Mohanan
相关产品推荐
相关产品推荐

