目录
这是一个基于合成数据的初步示例。希望不久之后能用真实研究的数据进行更新。
在只有一次预处理测量和一次后处理测量的情况下,我们可以使用类似于协方差分析(ANCOVA)的方法来分析非等效组设计(NEGD)实验的数据。基本模型是:
i指第i个样本,指给样本i的treatment,
是pre treatment,
是post treatment,
是一个截距项。非零值表示未经处理组随时间的变化,
是斜率。
是残差。
import matplotlib.pyplot as plt
import seaborn as sns
import causalpy as cp
%load_ext autoreload
%autoreload 2
%config InlineBackend.figure_format = 'retina'
seed = 42
生成合成数据
df = cp.load_data("anova1")
df.head()
让我们以两种不同的方式来可视化这些数据。
_, ax = plt.subplots(1, 2, figsize=(11, 5))
# ANCOVA plot
ax[0].axline((7, 7), (15, 15), color="k")
sns.scatterplot(x="pre", y="post", hue="group", alpha=0.5, data=df, ax=ax[0])
ax[0].set_title("ANCOVA style plot")
# Difference-in-difference plot
df_did = df.assign(unit=lambda x: x.index).melt(
id_vars=["group", "unit"], var_name="time", value_name="value"
)
sns.lineplot(
df_did,
x="time",
y="value",
hue="group",
units="unit",
estimator=None,
alpha=0.25,
ax=ax[1],
)
ax[1].set_title("Difference-in-difference style plot");
左侧的图对于进行类似协方差分析 (ANCOVA) 的分析最为相关。预处理测量和后处理测量分别位于 x 轴和 y 轴上。这张图展示了以下几点:
- 控制组的预处理测量值低于处理组,这与我们有非随机分配单位到处理组和控制组的想法一致。
- 控制组的点落在单位线上,表明控制组未受到处理的影响。
- 处理组在后处理得分上更高,这一点通过它们位于单位线之上的点显示出来。因此,从视觉上看,我们怀疑处理导致了结果指标的增加。
- 当然,可能有许多不同的结果模式,但我们可以通过同样的方法快速地对数据进行直观解释。
右侧的图展示的是完全相同的数据,但以一种清晰展示其与差分-in-差分 (Difference-in-Differences, DID) 分析方法相似的方式绘制。这里我们不再深入探讨差分-in-差分分析,更多相关信息可以参考差分-in-差分的例子。
分析
result = cp.pymc_experiments.PrePostNEGD(
df,
formula="post ~ 1 + C(group) + pre",
group_variable_name="group",
pretreatment_variable_name="pre",
model=cp.pymc_models.LinearRegression(sample_kwargs={"random_seed": seed}),
)
fig, ax = result.plot()
result.summary()
==================Pretest/posttest Nonequivalent Group Design=================== Formula: post ~ 1 + C(group) + pre Results: Causal impact = 1.9, $CI_{94%}$[1.7, 2.1] Model coefficients: Intercept -0.48, 94% HDI [-1.2, 0.23] C(group)[T.1] 1.9, 94% HDI [1.7, 2.1] pre 1, 94% HDI [0.98, 1.1] sigma 0.51, 94% HDI [0.46, 0.56]