介绍
目标
根据上面的最小二乘法代数推导过程,使用 Python 实现样本 x, y 的最小二乘法求解的函数。
def least_squares_function(x, y):
"""
参数:
x -- 样本 x 取值列表
y -- 样本 y 取值列表
返回:
w0 -- 线性方程参数, 保留两位小数
w1 -- 线性方程参数, 保留两位小数
"""
### TODO ###
return w0, w1 # 务必按此顺序返回
要求
- 题目需使用 Python3 完成,不能使用标准库及第三方库。
- 传入参数为列表类型,例如:x=[1,2,3,4]; y=[4,5,6,7]。
- 返回参数为四舍五入保留 2 位浮点数。
代码
def least_squares_function(x, y):
"""
参数:
x -- 样本 x 取值列表
y -- 样本 y 取值列表
返回:
w0 -- 线性方程参数, 保留两位小数
w1 -- 线性方程参数, 保留两位小数
"""
### TODO ###
sumxy=0
for i in x:
for j in y:
sumxy=sumxy+x*y
sumx=0
sumy=0
for i in x:
sumx=sumx+i
for i in y:
sumy=sumy+i
sumx2=0
for i in x:
sumx2=sumx2+i**2
w1=(len(x)*sumxy-(sumx*sumy))/(len(x)*sumx2-sumx**2)
w0=((sumx2*sumy)-(sumx*sumxy))/(len(x)*sumx2-sumx**2)
w1=round(w1,2)
w0=round(w0,2)
return w0, w1 # 务必按此顺序返回