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

Answer 2

这样的事情应该做

n = int(input('Enter n: '))
count = 1
sum = 0
while (count <= n):
  sum = sum + count*count
  print("{s}+".format(s=sum)      
  count = count + 1
print('Sum =', sum)