如何将列表中多余元素追加到交替合并后的结果末尾?
解决列表交替合并后追加剩余元素的问题
你的问题出在zip()函数的特性上——它只会遍历到两个列表中较短的那个的长度,超出部分的元素会被直接忽略,所以长列表的剩余元素没被加入结果。下面给你两种可行的解决方法:
方法一:手动处理剩余元素
在交替合并完成后,判断哪个列表更长,把超出的部分直接追加到结果里:
input1 = input("Enter numbers of the first list with spaces in between: ") a = input1.split() input2 = input("Enter numbers of the second list with spaces in between: ") b = input2.split() c = [] # 交替合并两个列表的共有部分 for x, y in zip(a, b): c += [x, y] # 追加较长列表的剩余元素 if len(a) > len(b): c.extend(a[len(b):]) elif len(b) > len(a): c.extend(b[len(a):]) print(c)
比如输入第一个列表1 2 3 4,第二个列表a b,结果会是['1', 'a', '2', 'b', '3', '4'],完美保留长列表的剩余元素。
方法二:用itertools.zip_longest简化代码
Python的itertools模块里的zip_longest会遍历到最长列表的长度,短列表的空缺位置会用默认的None填充,我们只需要过滤掉这些None即可:
from itertools import zip_longest input1 = input("Enter numbers of the first list with spaces in between: ") a = input1.split() input2 = input("Enter numbers of the second list with spaces in between: ") b = input2.split() c = [] for x, y in zip_longest(a, b): if x is not None: c.append(x) if y is not None: c.append(y) print(c)
这个方法不需要手动判断列表长度,代码更简洁,效果和方法一完全一致。
你之前用extend或append没效果,大概率是没在正确的时机使用——必须等zip循环结束后,再去处理长列表的剩余部分,而不是在循环内部操作。
内容的提问来源于stack exchange,提问作者Alex Kalampoukas
相关产品推荐
相关产品推荐

