python - 如何在它选择的 seaborn 分布图中添加一条垂直线？

Question

如何在 seaborn dist 图中 y 最大的 x 位置添加一条垂直线？

import seaborn as sns, numpy as np
sns.set(); np.random.seed(0)
x = np.random.randn(5000)
ax = sns.distplot(x, kde = False)

PS_ 在上面的例子中，我们知道它可能会选择0. 对于任何给定的 x 分布，我很想知道如何找到这个值。

标签： pythonmatplotlibplotseaborn

Answer 1

这是获得更准确点的一种方法。先得到平滑分布函数，用它来提取最大值，然后去掉。

import seaborn as sns, numpy as np
import matplotlib.pyplot as plt
sns.set(); np.random.seed(0)
x = np.random.randn(5000)
ax = sns.distplot(x, kde = True)

x = ax.lines[0].get_xdata()
y = ax.lines[0].get_ydata()
plt.axvline(x[np.argmax(y)], color='red')
ax.lines[0].remove()

编辑替代解决方案而不使用kde=True

import seaborn as sns, numpy as np
from scipy import stats
import matplotlib.pyplot as plt

sns.set(); np.random.seed(0)
x = np.random.randn(5000)
ax = sns.distplot(x, kde = False)

kde = stats.gaussian_kde(x) # Compute the Gaussian KDE
idx = np.argmax(kde.pdf(x)) # Get the index of the maximum
plt.axvline(x[idx], color='red') # Plot a vertical line at corresponding x

这会导致实际分布而不是密度值