Python中如何将列表内的None值替换为前后元素的平均值？

需求背景 #

需要处理列表中的None值：将每个None替换为其紧邻的前一个非None元素和后一个非None元素的平均值。举个例子：

• 输入：[1, None, 3, None, 7]
• 预期输出：[1, 2, 3, 5, 7]

现有问题 #

当前实现的substitution函数只能将None替换为固定索引的元素，无法动态计算前后元素的平均值，导致输出不符合预期。

测试用例 #

• 测试输入：[4, None, 6, 2, 3, None, 8, 11]
• 预期输出：[4, 5, 6, 2, 3, 5.5, 8, 11]

可行实现 #

下面是直接满足需求的Python函数，核心逻辑是为每个None定位前后有效的非None元素，计算平均值替换：

def replace_none_with_average(input_list):
    # 创建副本，避免修改原列表
    processed_list = input_list.copy()
    list_length = len(processed_list)
    
    for idx in range(list_length):
        if processed_list[idx] is None:
            # 寻找前一个非None元素
            prev_val = None
            for i in range(idx-1, -1, -1):
                if processed_list[i] is not None:
                    prev_val = processed_list[i]
                    break
            # 寻找后一个非None元素
            next_val = None
            for i in range(idx+1, list_length):
                if processed_list[i] is not None:
                    next_val = processed_list[i]
                    break
            # 计算平均值并替换（假设前后都存在有效元素，需处理边界可自行扩展）
            if prev_val is not None and next_val is not None:
                processed_list[idx] = (prev_val + next_val) / 2
    return processed_list

# 测试验证
test_input = [4, None, 6, 2, 3, None, 8, 11]
print(replace_none_with_average(test_input))
# 输出结果：[4, 5, 6, 2, 3, 5.5, 8, 11]

补充说明 #

• 上述代码默认列表中的None都有前后非None元素，如果需要处理边界情况（比如列表开头或结尾是None），可以添加判断逻辑：比如当只有前元素时用前元素填充，只有后元素时用后元素填充。
• 若存在连续多个None（比如[2, None, None, 6]），当前代码会将所有连续None都替换为前后有效元素的平均值，若需要实现线性插值效果，可进一步扩展逻辑。

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