时间戳

首页 > 解决方案 > 仅在 seaborn 的最后一张图中显示的平均中值模式线

python - 仅在 seaborn 的最后一张图中显示的平均中值模式线

问题描述

我试图在两个图中显示mean、median和mode线，但它们仅在最后一个图中可见：

#Cut the window in 2 parts
f, (ax_box, ax_hist) = plt.subplots(2, sharex=True, gridspec_kw={"height_ratios": (0.2, 1)})
#plt.figure(figsize=(10,7));
mean=df[' rating'].mean()
median=df[' rating'].median()
mode=df[' rating'].mode().get_values()[0]
plt.axvline(mean, color='r', linestyle='--')
plt.axvline(median, color='g', linestyle='-')
plt.axvline(mode, color='b', linestyle='-')
plt.legend({'Mean':mean,'Median':median,'Mode':mode})

sns.boxplot(df[" rating"], ax=ax_box)
sns.distplot(df[" rating"], ax=ax_hist)

ax_box.set(xlabel='')

标签： pythonmatplotlibseaborn

解决方案

该命令plt使用当前轴，而不是所有定义的轴。要在特定轴上绘制某些东西，您必须告诉 matplotlib/seaborn，您的意思是哪个轴：

from matplotlib import pyplot as plt
import pandas as pd
import seaborn as sns

df = pd.DataFrame({" rating": [1, 2, 3, 4, 6, 7, 9, 9, 9, 10], "dummy": range(10)})

f, (ax_box, ax_hist) = plt.subplots(2, sharex=True, gridspec_kw= {"height_ratios": (0.2, 1)})
mean=df[' rating'].mean()
median=df[' rating'].median()
mode=df[' rating'].mode().values[0]

sns.boxplot(data=df, x=" rating", ax=ax_box)
ax_box.axvline(mean, color='r', linestyle='--')
ax_box.axvline(median, color='g', linestyle='-')
ax_box.axvline(mode, color='b', linestyle='-')

sns.histplot(data=df, x=" rating", ax=ax_hist, kde=True)
ax_hist.axvline(mean, color='r', linestyle='--', label="Mean")
ax_hist.axvline(median, color='g', linestyle='-', label="Median")
ax_hist.axvline(mode, color='b', linestyle='-', label="Mode")

ax_hist.legend()

ax_box.set(xlabel='')
plt.show()

样本输出： 在此处输入图像描述

如果你有一大堆子图，你可以循环处理这个任务：

f, bunch_of_axes = plt.subplots(200)
...
for ax in bunch_of_axes:
    ax.axvline(mean, color='r', linestyle='--')
    ax.axvline(median, color='g', linestyle='-')
    ax.axvline(mode, color='b', linestyle='-')

2021 年更新：我更改了 pandas 代码，因为get_values()现在已弃用。Seaborn 也已弃用distplot。替代方案是displot，一个没有 ax 参数的图形级函数，或者histplot其行为与 略有不同distplot。

我现在在另一个线程中总结了如何distplot使用histplot.

推荐阅读