gpt4 book ai didi

python - Statsmodels:编写公式的简短方法

转载 作者:行者123 更新时间:2023-12-04 12:47:13 27 4
gpt4 key购买 nike

使用状态模型的逻辑回归模型:

log_reg = st.logit(formula = 'label ~ pregnant + glucose + bp + insulin + bmi + pedigree + age', data=pima).fit()

公式的第二部分(怀孕 + 葡萄糖 + bp + 胰岛素 + bmi + 血统 + 年龄)有什么简短的写法吗?这里必须明确提及所有列。如果超过100列,就很难写,而且语句会很长。

最佳答案

如果 df 是 pd.DataFrame,而 y 是目标变量,则此函数返回您要查找的公式字符串。

def formula_from_cols(df, y):
return y + ' ~ ' + ' + '.join([col for col in df.columns if not col==y])

关于python - Statsmodels:编写公式的简短方法,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/35518477/

27 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com