编写可指定取数组前N个元素进行求和的代码需求
实现指定数量的数组前n项求和
方法一:for循环实现(优先方案)
这种方法逻辑直观,能适配任意指定数量,无需硬编码索引偏移:
x = [1, 3, 4, 2, 6, 9, 4] amount = 3 # 可替换为任意需要的数量,比如4 sum_result = 0 for i in range(amount): sum_result += x[i] print(sum_result) # amount=3时输出8,amount=4时输出11
通过range(amount)生成0到amount-1的索引,逐个取出数组元素累加到总和,完全适配不同数量需求。
方法二:内置函数+切片(简洁方案)
利用Python切片和内置sum()函数,一行代码就能完成需求,是更简洁的Python风格写法:
x = [1, 3, 4, 2, 6, 9, 4] amount = 4 sum_result = sum(x[:amount]) print(sum_result) # 输出11
x[:amount]直接截取数组前amount个元素,sum()函数对截取后的子数组直接求和,代码简洁高效。
方法三:while循环实现
如果想换一种循环逻辑,while循环也能实现相同功能:
x = [1, 3, 4, 2, 6, 9, 4] amount = 3 sum_result = 0 index = 0 while index < amount: sum_result += x[index] index += 1 print(sum_result) # 输出8
初始化索引为0,当索引小于指定数量时累加对应元素,索引自增直到达到指定数量后停止循环。
内容的提问来源于stack exchange,提问作者PoorPythonProgrammer
相关产品推荐
相关产品推荐

