问题描述
嗨,我想将一项 A/B 测试的结果可视化。该实验跟踪 4 个指标,我想在一个图中显示它们。我的数据框的架构是:
test_control |公制1 |公制2 |公制3 |公制4
有谁知道如何通过 matplotlib、pandas 或 seaborn 进行绘图?
提前致谢!
解决方法
我发现在 R 中可能更容易完成。 在python中,我计算了误差条,然后使用matplotlib.pyplot.errorbar来绘制:
获取 CI
kpi_map = {'kpi':[],'mean_diff':[],'err':[],'pval':[]}
for col in metrics:
sp1 = df.loc[df['test_control']=='test'][col]
sp2 = df.loc[df['test_control']=='control'][col]
std1 = np.std(sp1,ddof=1)
std2 = np.std(sp2,ddof=1)
mean_diff_std = (std1**2/len(sp1) + std2**2/len(sp2)) **0.5
mean_diff = sp1.mean() - sp2.mean()
kpi_map['kpi'].append(col)
kpi_map['mean_diff'].append(mean_diff)
kpi_map['err'].append(1.96*mean_diff_std)
情节
df_kpi = pd.DataFrame(data = kpi_map)
plt.errorbar(y=df_kpi['kpi'],x=df_kpi['mean_diff'],xerr=df_kpi['err'],fmt='o',elinewidth=2,capsize=4,capthick=2)