#导入库
import pandas as pd
import numpy as np
#导入数据
data=pd.read_excel("C:\\Users\\Administrator\\Desktop\\1599639998_298148.xlsx")
from sklearn import linear_model
#将没有求得的斜率预设为缺失值
data["斜率"]=np.nan
#求数据行数
rows=data.shape[0]
#从第一行(第一个人)开始,进行循环,多少行数据就循环多少次,做多少个回归
for i in range(rows):
reg=linear_model.LinearRegression()
#调取第i行的x数据
id=data["序号"][i]
x=data.loc[i,"近6":"近1"].values.reshape(6,1)
y=[1,2,3,4,5,6]
#如果遇到不能回归的情况可以跳过去,比如缺失值太多无法回归
try:
reg.fit(x,y)
coef=reg.coef_[0]
except:
coef="异常情况"
data.loc[i,"斜率"]=coef
#将结果导出
data.to_excel("C:\\Users\\Administrator\\Desktop\\结果.xlsx")
