我运行了一个线性回归模型并得到了系数。如何在系数旁边打印变量?
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
3条答案
按热度按时间kq0g1dla1#
我的首选方法是
pd.Series(reg.coef_,index=df.columns)
,然后打印是免费的。而且它更容易与pd.Series
工作的其他计算,通过pd.concat
的模型比较等。5cnsuln72#
如果你只想把变量打印在系数旁边,你可以这样写:
tnkciper3#
我相信
'\n'.join(col+' coef: '+str(coef) for (col, coef) in zip(df.columns, reg.coef_))
可以满足您的要求。但是,我推荐以下代码,因为它提供的输出更容易阅读:另外,所有这些都假设你的列名已经是字符串了,如果不是,你应该用
str(col)
替换col
。