使用python statsmodels修復summary_col中的標簽外生變量

Question

我想生成類似於summary_col（標准期刊表）生成的回歸表，但帶有自定義解釋變量標簽。

有沒有辦法更改保存在模型參數屬性中的行名稱？

到目前為止，我將變量重命名為最接近我想要的，但應該有更好的方法來做到這一點。

Answer 1

假設你已經完成

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

我建議 (1) 有一個字典來保存所有的重新標記： dic = {original_vname: new_name} a (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 后，變量將顯示新標簽。