如何在seaborn.distplot中填充不同颜色的区域

时间:2017-10-11 10:08:16

标签: python matplotlib seaborn

是否可以用两个阈值线(line1和line2)之外的区域填充颜色,并通过distplot绘制的KDE曲线限制在Y轴上?(表示我的应用程序为3-sigmas)

import pylab as pl
import seaborn as sns
#plotting the two lines
p1 = pl.axvline(x=line1,color='#EF9A9A')
p2 = pl.axvline(x=line2,color='#EF9A9A')
#plotting the PDF
sns.distplot(stat, hist=True,color='#388E3C')

result figure

1 个答案:

答案 0 :(得分:4)

您可以使用fill_between填充曲线下方的区域。要从seaborn图中访问KDE曲线,您可以先绘制一条曲线,这样ax.lines只有一个元素,这是感兴趣的曲线。其数据通过kde_x, kde_y = ax.lines[0].get_data()获得。

然后使用ax.fill_between()允许填充曲线下方的区域。要将此限制在某个给定的数据范围之外,可以使用where关键字参数(并且应设置interpolate=True以使该区域上升到相关点。)

ax.fill_between(kde_x, kde_y, where=(kde_x<x0) | (kde_x>x1) , 
                interpolate=True, color='#EF9A9A')

完整示例:

import numpy as np
import matplotlib.pyplot as plt
import seaborn as sns

stat=np.random.randn(100)
x0 = -1
x1 = 1

#plotting the PDF (do this before plotting anything else)
ax = sns.distplot(stat, hist=True,color='#388E3C')
kde_x, kde_y = ax.lines[0].get_data()

#plotting the two lines
p1 = plt.axvline(x=x0,color='#EF9A9A')
p2 = plt.axvline(x=x1,color='#EF9A9A')


ax.fill_between(kde_x, kde_y, where=(kde_x<x0) | (kde_x>x1) , 
                interpolate=True, color='#EF9A9A')

plt.show()

Image produced by the above code

<小时/> 最初问题的旧答案:

您可以使用axvspan,从左边的x限制开始,到第一行的位置,另一个从第二行的位置开始,然后转到右边的x限制。

import numpy as np
import matplotlib.pyplot as plt
import seaborn as sns

stat=np.random.randn(100)
x0 = -1
x1 = 1

#plotting the two lines
p1 = plt.axvline(x=x0,color='#EF9A9A')
p2 = plt.axvline(x=x1,color='#EF9A9A')

#plotting the PDF
ax = sns.distplot(stat, hist=True,color='#388E3C')

xlim = ax.get_xlim()
ax.axvspan(xlim[0], x0, color='#EF9A9A', alpha=0.5)
ax.axvspan(x1, xlim[1], color='#EF9A9A', alpha=0.5)
#reset xlim
ax.set_xlim(xlim)
plt.show()

Image produced by the above code

在这里,我们需要在设置跨度后调整xlimits;原因是,在跨度到位的情况下,自动缩放会在轴的两端添加另外5%的填充,从而产生空白区域。或者,您可以为x轴使用零边距,ax.margins(x=0)