目录
生成合成数据
分析
这是一个基于合成数据的初步示例。希望不久之后能用真实研究的数据进行更新。
在只有一次预处理测量和一次后处理测量的情况下，我们可以使用类似于协方差分析(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]
