【问题标题】:Is there a way to display regression coefficients in a pandas data frame for categorical independent variables?有没有办法在熊猫数据框中显示分类自变量的回归系数?
【发布时间】:2021-10-19 23:59:43
【问题描述】:

我已经建立了一个多元线性回归模型,并使用model.coef_ 找到了系数。 我想制作一个显示每个因素及其系数的 pandas 数据框。

pd.DataFrame(model.coef_, x.columns, columns = ['coef']).sort_values(by = 'coef', ascending = False) 仅适用于数值自变量。我有两列分类变量,我对它们都进行了编码。

假设一列的两个值是“男性”和“女性”,我想显示单个系数,如

coef
Male 0.2
Female 0.3

可以这样做吗?

【问题讨论】:

    标签: python pandas regression coefficients


    【解决方案1】:

    之后可以set_index

    (pd.DataFrame({'coef':model.coef_, 'category':x.columns})
    .sort_values(by = 'coef', ascending = False)
    .set_index('category'))
    

    【讨论】:

    • 我收到一个错误。 AttributeError: columns not found
    • 是否有一个名为 x 的数据框有列?根据您的问题,似乎有一个。
    • 有没有可能你覆盖了 x 是什么。当你打电话给type(x)时你会得到什么
    • scipy.sparse.csr.csr_matrix 我认为它在编码时已更改。这是我的代码ct = ColumnTransformer(transformers=[('encoder', OneHotEncoder(), [0,1])], remainder='passthrough') X = ct.fit_transform(X)
    • 它在转换回 pandas 框架并将系数展平为一维 numpy 数组后工作。谢谢:)
    猜你喜欢
    • 1970-01-01
    • 1970-01-01
    • 2020-09-23
    • 2021-02-26
    • 2021-12-10
    • 1970-01-01
    • 2019-07-12
    • 2021-04-10
    • 2021-11-30
    相关资源
    最近更新 更多