Python3求助：实现每个x值对应≥该x值的y值之和计算并生成等长列表

功能需求 #

• 遍历x_values中的每个元素作为固定x值，计算所有对应索引的x值≥该固定x值的y_values元素之和
• 输出列表的长度必须与x_values完全一致

示例 #

输入：

x_values = [1,2,3,3,4,5]
y_values = [9,4,3,1,2,1]

输出：

[20,11,7,7,3,1]

原代码问题分析 #

你写的代码存在三个核心问题：

1. my_list未在外层循环后清空，导致每次求和都是累加之前所有符合条件的y值，结果持续增大
2. my_list2.append(sum(my_list))放在内层循环中，每次满足条件就添加一次求和结果，最终my_list2长度远超x_values
3. 逻辑偏差：应该为每个固定x值计算一次总和，而非逐步累加后多次记录

修正后的代码 #

基础循环版 #

x_values = [1,2,3,3,4,5]
y_values = [9,4,3,1,2,1]

result = []
for j in range(len(x_values)):
    current_sum = 0
    # 针对当前固定x值，遍历所有元素计算符合条件的y值总和
    for i in range(len(x_values)):
        if x_values[i] >= x_values[j]:
            current_sum += y_values[i]
    result.append(current_sum)

print(result)  # 输出: [20, 11, 7, 7, 3, 1]

简洁列表推导式版 #

x_values = [1,2,3,3,4,5]
y_values = [9,4,3,1,2,1]

# 用zip配对x和y值，遍历每个固定x值计算总和
result = [sum(y for x, y in zip(x_values, y_values) if x >= fixed_x) for fixed_x in x_values]

print(result)  # 输出: [20, 11, 7, 7, 3, 1]

修正说明 #

• 外层循环每次初始化current_sum为0，确保每个固定x值的求和独立计算
• 将求和结果的添加操作放在外层循环末尾，保证输出列表长度与x_values一致
• 列表推导式版本通过zip将x和y值一一配对，代码更简洁易读

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