首页 > 解决方案 > 如何在 Python 中显示每个单独变量的相关系数?

问题描述

我运行了一个线性回归模型并得到了我的系数。如何在系数旁边打印变量?

df = pd.read_csv('data', sep=";")
reg = linear_model.LinearRegression()
reg.fit(df[["age", "area", "bedrooms"]],df.price)
print(reg.coef_)

Output 

[ 6.55199614e-02 -1.86317709e+00  2.20902007e-02]

我希望输出是

age coef: 6.55199614e-02
area coef: -1.86317709e+00
bedroom coef: 2.20902007e-02

标签: pythonpandas

解决方案


我的首选方法是pd.Series(reg.coef_,index=df.columns),然后免费打印。此外,它更容易pd.Series用于其他计算、模型比较pd.concat等。


推荐阅读