我有一个pandas
数据框,其中包含字段x
,y
,year
。 year
的值为[-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'])
。 (非常轻的)文档没有提到这一点,但它似乎是我真正应该支持的核心功能。