Python如何合并列表指定连续元素并排查代码错误

原有代码问题点 #

• 字符串拼接时未添加空格分隔，直接用+拼接相邻字符串，导致出现CatAnd这类不符合格式的结果
• 合并范围判断错误：遇到Cat时仅向后取1个元素拼接，没有把连续跟随的And、Mouse两个元素全部纳入合并
• 迭代器推进步长错误：仅调用1次next()跳过1个后续元素，导致本该被合并的Mouse被当作独立元素迭代输出
• 缺少边界判断：如果Cat出现在列表末尾不足2个后续元素的位置，直接调用next()会触发StopIteration运行时错误

可运行修正代码 #

最易读、易维护的索引遍历实现：

my_string = ['Dog', 'Cat', 'And', 'Mouse', 'Whale', 'Cat', 'And', 'Mouse']
new_string = []
idx = 0
list_len = len(my_string)

while idx < list_len:
    # 匹配连续的['Cat', 'And', 'Mouse']片段
    if idx + 2 < list_len and my_string[idx] == 'Cat' and my_string[idx+1] == 'And' and my_string[idx+2] == 'Mouse':
        new_string.append('Cat And Mouse')
        idx += 3  # 跳过已合并的3个元素
    else:
        new_string.append(my_string[idx])
        idx += 1

print(new_string)

运行输出完全符合预期：

['Dog', 'Cat And Mouse', 'Whale', 'Cat And Mouse']

如果偏好迭代器写法，可参考如下实现：

my_string = ['Dog', 'Cat', 'And', 'Mouse', 'Whale', 'Cat', 'And', 'Mouse']
it = iter(my_string)
new_string = []

for item in it:
    if item == 'Cat':
        # 依次取出后续两个元素，拼接时补空格
        new_string.append(f"{item} {next(it)} {next(it)}")
    else:
        new_string.append(item)

内容的提问来源于stack exchange，提问作者swimmeryay15