ermutuxia 回复您的文章《用python拟合出每个客户的线性方程，主要想看斜率》

2020-09-10 13:36

ermutuxia 回复您的文章《用python拟合出每个客户的线性方程，主要想看斜率》

#导入库

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]

#要求y是一维数组

y=data.loc[i,"近6":"近1"].values

#要求x是二维数组，哪怕x只由一个变量

x=np.array([1,2,3,4,5,6]).reshape(6,1)

try:

reg.fit(x,y)

coef=reg.coef_[0]

except:

coef="异常情况"

data.loc[i,"斜率"]=coef

2020-09-10 13:35

ermutuxia 赞了您的文章 《python中如何填补缺失值？》

2020-09-10 12:39

Bena0606 赞了您的文章 《电商案例之客户价值模型》

2020-09-10 12:06