使用python statsmodels修复summary_col中的标签外生变量

Question

我想产生像summary_col（标准日记帐表）产生的那些回归表，但是带有自定义的解释变量标签。

是否可以更改保存在模型params属性中的行名？

到目前为止，我将变量重命名为最接近我想要的变量，但是应该有一种更好的方法。

Answer 1

假设您已完成

reg = smf.ols(formula = "y~x1+x2+x3").fit()

我建议（1）拥有一本字典，其中存放所有重新标记：dic = {原始vname：new_name}（2）一对有用的功能：

def rename_vars(vname): 
    to_ret = vname
    for orig_vname in list(dic.keys()):
        if vname == 'original_vname':
            to_ret = dic['original_vname'] 
    return to_ret

和

def rename_ols(reg): 
    for i in range(len(reg)): 
        reg[i] = rename_vars(reg[i])

然后，只需执行：

rename_ols(reg.model.exog_names) 

就是这样。调用summary_col后，变量将显示新标签。