计算列表元素及其邻居之和时新列表长度不符的修复方法

我有一个生成随机列表的代码，会创建一个长度在3到21之间、元素为1-100随机数的列表：

import random
list1 = []
start = 1
end = random.randint(3, 21)
for i in range(start, end + 1):
    list1.append(random.randrange(1, 101, 1))
print("My list length is", end)
print("List1 (Original List):", list1)

我的需求是创建一个和list1长度相同的新列表，规则是：

• 首元素：自身 + 右邻居
• 尾元素：自身 + 左邻居
• 中间元素：左邻居 + 自身 + 右邻居

我尝试写了这段代码，但输出的新列表长度总是和list1不一样：

list2 = ( [list1[0] + list1[1]] + [list1[i-1] + list1[i] + list1[i+1] for i in range(1, len(list1)-2)] + [list1[-2] + list1[-1]] )
print(list2)

请问该怎么修复这个问题？

问题分析 #

你的代码问题出在列表推导式的range范围上：range(1, len(list1)-2)会让循环只运行到索引len(list1)-3（因为range是左闭右开的）。举个例子，如果原列表长度是5，len(list1)-2等于3，所以range(1,3)只会遍历i=1这一个索引，加上首尾两个元素，总长度是1+2=3，比原列表的5少了2个，这就是长度不一致的核心原因。

修复方案 #

这里有两种简单的修复方式，选哪种都能满足需求：

方式1：修正列表推导式的range范围 #

把range(1, len(list1)-2)改成range(1, len(list1)-1)，这样中间的列表推导式会覆盖从索引1到len(list1)-2的所有中间元素，再加上首尾两个元素，总长度就和原列表完全一致了：

list2 = [list1[0] + list1[1]] + [list1[i-1] + list1[i] + list1[i+1] for i in range(1, len(list1)-1)] + [list1[-2] + list1[-1]]

方式2：用循环遍历每个索引（更直观易读） #

如果觉得列表推导式的范围容易搞混，不如直接遍历每个索引，根据位置判断计算逻辑——这样代码更清晰，后续维护也不容易出错：

list2 = []
for i in range(len(list1)):
    if i == 0:
        # 首元素：自身+右邻居
        list2.append(list1[i] + list1[i+1])
    elif i == len(list1)-1:
        # 尾元素：自身+左邻居
        list2.append(list1[i] + list1[i-1])
    else:
        # 中间元素：左+自身+右
        list2.append(list1[i-1] + list1[i] + list1[i+1])

你可以测试一下，两种方式都能生成和原列表长度完全匹配的新列表，完美符合你的需求。

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