График результатов модели смешанного эффекта по группам в Python

Question

Я использую модель смешанного эффекта со statsmodel в python и не могу найти способ отобразить результаты по групповому уровню. Моя цель - добраться до такого рода сюжетов:

, что возможно в R .

Для справки: в моем фрейме данных df у меня есть три столбца = metric, experiment_name, country.

. Этот код я использую

import statsmodels.api as sm
import statsmodels.formula.api as smf

formula = "metric ~ experiment_name"

md = smf.mixedlm(formula, df, groups=df["country"])
mdf = md.fit()
print(mdf.summary())

. Главная проблема в том, что я не могу понять, как получить результаты на уровне группы. Я много гуглил без удачи.

Answer 1

Я нашел способ получить результаты на уровне группы, по крайней мере, для случайных эффектов.

mdf.random_effects

и построить для него график леденцов с matplotlib после преобразования данных к кадру данных и транспонирования.