Python中的条件回归模型
Python中的条件回归模型
提问于 2017-07-12 23:12:42
在python中，我使用以下代码创建了一个回归模型：
import statsmodels.formula.api as sm
df = pd.DataFrame({"A": data['dur'], "B": data['agreement'], "C": data['vidPersonIdeoDiff'], "D": data['opinionChange'], "G": data['fake']})
result = sm.ols(formula="A ~ B + C + D + G", data=df).fit()
print(result.params)
print(result.summary())现在，我感兴趣的是向同一回归添加一个条件或“if-statement”。我如何运行它，但只针对特定的组，例如如果年龄< 36岁？在Stata中，等效项为：
reg dur agreement vidPersonIdeoDiff opinionChange fake if age < 36你知道如何在python中复制stata代码行吗？如果解决方案与我一直使用的statsmodels.formula.api模块一致，这将特别有帮助。谢谢!
Stack Overflow用户
发布于 2017-07-12 23:21:14
假设'age‘是你有条件的变量：
result = sm.ols(formula="A ~ B + C + D + G", data=df[data['age']<36]).fit()页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接：
https://stackoverflow.com/questions/45061576
复制相关文章
相似问题
腾讯云开发者
