Python 中的线性回归系数

Question

我是 Python 新手，我正在尝试使用 csv 进行线性回归，我需要获取系数，但我不知道如何获取。 这是我尝试过的：

import statsmodels.api as sm
x = datos1['Ozone']
y = datos1['Temp']
x = np.array(x)
y= np.array(y)
model = sm.OLS(y, x)
results = model.fit()
print(results.summary())

你可以帮帮我吗？ 谢谢。

Answer 1

尝试跑步，

import statsmodels.api as sm
x = datos1['Ozone']
y = datos1['Temp']
x = np.array(x)`enter code here`
y= np.array(y)
model = sm.OLS(y, x)
results = model.fit()

print(results.params)

详情请参见： statsmodels.regression.linear_model.OLS

Answer 2

有两种方法可以获取OLS模型的参数。

方式一：使用model.params

x = datos1['Ozone']
y = datos1['Temp']
x = np.array(x)
y= np.array(y)
model = sm.OLS(y, x)
results = model.fit()
print(results.params)

方式 2：将model.summary()与 pandas 一起使用

x = datos1['Ozone']
y = datos1['Temp']
x = np.array(x)
y= np.array(y)
model = sm.OLS(y, x)
results = model.fit()

df = pd.read_html(results.summary().tables[1].as_html(),header=0,index_col=0)[0]
print(df.columns.values) # get parameters as dataframe 
print(df['coef']) # get coefficients