python:如何使用增量数对序列求和

Question

n = int(input('Enter n: '))
count = 1
sum = 0
number = 1
while (count <= n):
    sum = sum + number
    count = count + 1
    number = number + 2
print('Sum =', sum)

是否可以对 1 + 4 + 9 + 16 + 25 + 36 + 49 + 64 .... + n 使用相同的概念

Answer 1

您可以使用列表理解来使其更加优雅和Pythonic:

def sum_series(start, end):
  return sum([i**2 for i in range(start, end+1)])

print(sum_series(1,10))

输出:

385

或者使用高阶函数:

>>> sum(map(lambda x: x**2, range(1,11)))
385