带有单独数据点的pyplot条形图

Question

我有一个对照组和治疗组的数据。 matplotlib是否能够创建条形图，其中条形高度是每个组的平均值与该组中的各个数据点重叠的值？我想可视化实际数据点的分布，类似于显示的here。

我曾经考虑过结合使用箱线图和散点图，但是我的尝试没有成功。

Answer 1

这是一种完全按照您所说的解决方案：用散点图覆盖条形图。

当然，您可以进一步调整图的位置：散点图的图标题，轴标签，颜色，宽度，标记形状...

import matplotlib.pyplot as plt
np.random.seed(123)

w = 0.8    # bar width
x = [1, 2] # x-coordinates of your bars
colors = [(0, 0, 1, 1), (1, 0, 0, 1)]    # corresponding colors
y = [np.random.random(30) * 2 + 5,       # data series
    np.random.random(10) * 3 + 8]

fig, ax = plt.subplots()
ax.bar(x,
       height=[np.mean(yi) for yi in y],
       yerr=[np.std(yi) for yi in y],    # error bars
       capsize=12, # error bar cap width in points
       width=w,    # bar width
       tick_label=["control", "test"],
       color=(0,0,0,0),  # face color transparent
       edgecolor=colors,
       #ecolor=colors,    # error bar colors; setting this raises an error for whatever reason.
       )

for i in range(len(x)):
    # distribute scatter randomly across whole width of bar
    ax.scatter(x[i] + np.random.random(y[i].size) * w - w / 2, y[i], color=colors[i])

plt.show()

它将产生此图

Answer 2

这里是使用Seaborn的解决方案，与直接使用Matplotlib相比，它的代码更短，但具有一定的灵活性：

import seaborn as sns, matplotlib.pyplot as plt
sns.set(style="whitegrid")

tips = sns.load_dataset("tips")

sns.barplot(x="day", y="total_bill", data=tips, capsize=.1, ci="sd")
sns.swarmplot(x="day", y="total_bill", data=tips, color="0", alpha=.35)

plt.show()

结果如下：