如何在大Pandas中建立两个固定效应和一个随机效应的混合线性模型?

2024-03-29 13:12:34 发布

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

新的编码,所以我甚至不知道如何去做这件事
我有一个类似这样的df:

        ID        A          B
0       1         20         4.0
1       1         24         5.0
2       2         24         0.0
3       2         9          2.0
4       3         24         1.0
5       3         24         0.0

我想做一个混合线性模型,让我把a和B关联起来,以显示ID中的可变性。a和B是我的固定效应,而ID是我的随机效应。我希望A依赖于B。
我试图使用此代码来完成此操作:

import statsmodels.api as sm
model = sm.MixedLM.from_formula("B~ A", df1, groups=df1["ID"])
result = model.fit()
print(result.summary())

然而,我不断得到无效的语法错误
我还想知道如何在散点图上绘制这个,因为我想根据每个ID的每条线的斜率来找出a和B之间的相关性


Tags: 代码模型importapiid编码dfmodel