利用Python实现自然数平方和

# 以下代码实现1^2+2^2+...+n^2

import numpy as np
print("请输入一个整数")
n = int(input("n="))
a_n = np.arange(n+1) # 表示一个数列
sum = 0 # 用于存储数列a_n前n项的平方和
for i in a_n:
    sum = sum + a_n[i] ** 2
    i = i + 1
print(sum)