Statsmodels:按子组的回归

2024-04-25 15:09:05 发布

您现在位置:Python中文网/ 问答频道 /正文

我有一个pandas数据帧,它包含字段xyyearyear具有值[-100, 100, 1999]。我想按年份创建子组并运行reg x y。你知道吗

我知道我可以通过类似的方法在熊猫方面解决这个问题

for y in df.year.unique():
    smf.ols('y ~ x', data=df.loc[df.year == y])

我想知道在statsmodels中是否有更好的方法来实现这一点,比如smf.ols('y ~ x', data=df, by=['year'])。(非常轻的)文档没有提到这一点,但它似乎是我真正应该支持的一个核心特性。你知道吗


Tags: 数据方法inpandasdffordatareg