数据分析数据分析

详解seaborn图表中的kdeplot、rugplot、dis

2020-09-04  本文已影响0人  python大数据分析

一、seaborn简介

seaborn是Python中基于matplotlib的具有更多可视化功能和更优美绘图风格的绘图模块，当我们想要探索单个或一对数据分布上的特征时，可以使用到seaborn中内置的若干函数对数据的分布进行多种多样的可视化。

本文以jupyter notebook为编辑工具，针对seaborn中的kdeplot、rugplot、distplot和jointplot，对其参数设置和具体用法进行详细介绍。

二、kdeplot

seaborn中的kdeplot可用于对单变量和双变量进行核密度估计并可视化，其主要参数如下：

下面我们来看几个示例来熟悉kdeplot中上述参数的实际使用方法：

首先我们需要准备数据，本文使用seaborn中自带的鸢尾花数据作为示例数据，因为在jupyter notebook中运行代码，所以加上魔术命令%matplotlib inline使得图像得以在notebook中显示。

import seaborn as sns
sns.set(color_codes=True)
import matplotlib.pyplot as plt
%matplotlib inline
#加载seaborn自带的鸢尾花数据集，格式为数据框
iris = sns.load_dataset('iris')
#分离出setosa类的花对应的属性值
setosa = iris.loc[iris.species == "setosa"].reset_index(drop=True)
#分离出virginica类的花对应的属性值
virginica = iris.loc[iris.species == "virginica"].reset_index(drop=True)

首先我们不修改其他参数只传入数据来观察绘制出的图像：

#绘制iris中petal_width参数的核密度估计图
ax = sns.kdeplot(iris.petal_width)

加上红色填充颜色，并禁止图例显示：

ax = sns.kdeplot(iris.petal_width,shade=True,color='r')

修改为核密度分布：

ax = sns.kdeplot(iris.petal_width,
                 shade=True,
                 color='r',
                 cumulative=True)

交换x-y轴位置：

ax = sns.kdeplot(iris.petal_width,
                 shade=True,
                 color='r',
                 vertical=True)

下面我们来绘制双变量联合核密度估计图：

#绘制setosa花的petal_width与petal_length的联合核密度估计图
ax = sns.kdeplot(setosa.petal_width,
                 setosa.petal_length)

修改调色方案为蓝色，并设置shade_lowest=True：

ax = sns.kdeplot(setosa.petal_width,
                 setosa.petal_length,
                 cmap='Blues',
                 shade=True,
                 shade_lowest=True)

在上图基础上修改shade_lowest=False：

ax = sns.kdeplot(setosa.petal_width,
                 setosa.petal_length,
                 cmap='Blues',
                 shade=True,
                 shade_lowest=False)

可以看到这时最低密度估计曲线之外的区域没有被调色方案所浸染。

将核密度曲线区间个数修改为5：

ax = sns.kdeplot(setosa.petal_width,
                 setosa.petal_length,
                 cmap='Blues',
                 shade=True,
                 shade_lowest=False,
                 n_levels=5)

可以看到这时的核密度区间要粗略很多。

在同一个子图中绘制两个不同一维总体的核密度估计图，这里为了把它们区分开分别定义了label参数以显示在图例中：

ax1 = sns.kdeplot(setosa.petal_width,label='setosa.petal_width')
ax2 = sns.kdeplot(virginica.petal_width,label='virginica.petal_width')

在同一个子图中绘制两个不同二维总体的核密度估计图：

ax1 = sns.kdeplot(setosa.sepal_width,setosa.sepal_length,
                  cmap='Blues',
                  shade=True,
                  shade_lowest=False)
ax2 = sns.kdeplot(virginica.sepal_width,virginica.sepal_length,
                  cmap='Greens',
                  shade=True,
                  shade_lowest=False)

三、rugplot

rugplot的功能非常朴素，用于绘制出一维数组中数据点实际的分布位置情况，即不添加任何数学意义上的拟合，单纯的将记录值在坐标轴上表现出来，相对于kdeplot，其可以展示原始的数据离散分布情况，其主要参数如下：

使用默认参数进行绘制：

ax = sns.rugplot(iris.petal_length)

调换所处的坐标轴：

ax = sns.rugplot(iris.petal_length,axis='y')

修改小短条高度和颜色：

ax = sns.rugplot(iris.petal_length,
                 color='r',
                 height=0.2)

四、distplot

seaborn中的distplot主要功能是绘制单变量的直方图，且还可以在直方图的基础上施加kdeplot和rugplot的部分内容，是一个功能非常强大且实用的函数，其主要参数如下：

使用默认参数进行绘制：

ax = sns.distplot(iris.petal_length)

修改所有对象的颜色，绘制rugplot部分，并修改bins为20：

ax = sns.distplot(iris.petal_length,color='r',
                 rug=True,
                 bins=20)

在上图的基础上强行拟合卡方分布并利用参数字典设置fit曲线为绿色：

from scipy.stats import chi2                
ax = sns.distplot(iris.petal_length,color='r',
                 rug=True,
                 bins=20,
                 fit=chi2,
                 fit_kws={'color':'g'})

修改norm_hist参数为False使得纵轴显示的不再是密度而是频数（注意这里必须关闭kde和fit绘图的部分，否则纵轴依然显示密度），利用hist_kws传入字典调整直方图部分色彩和透明度，利用rug_kws传入字典调整rugplot部分小短条色彩：

ax = sns.distplot(iris.petal_length,color='r',
                 rug=True,
                 kde=False,
                 bins=20,
                 fit=None,
                 hist_kws={'alpha':0.6,'color':'orange'},
                 rug_kws={'color':'g'},
                 norm_hist=False)

五、jointplot

之所以按照kdeplot-rugplot-distplot的顺序来介绍是因为distplot中涉及到kdeplot与rugplot中的相关内容，而本文最后要介绍的函数jointplot中聚合了前面所涉及到的众多内容，用于对成对变量的相关情况、联合分布以及各自的分布在一张图上集中呈现，其主要参数如下：

在默认参数设置下绘制成对变量联合图：

ax = sns.jointplot(x='sepal_length',y='sepal_width',data=setosa)

值得一提的是，jointplot还贴心的在图像上说明了成对变量之间的皮尔逊简单相关系数以及相关性检验的p值结果。

将kind参数设置为'reg'，为联合图添加线性回归拟合直线与核密度估计结果：

ax = sns.jointplot(x='sepal_length',y='sepal_width',data=setosa,
                  kind='reg')

修改kind为'hex'来为联合图生成六边形核密度估计：

ax = sns.jointplot(x='sepal_length',y='sepal_width',data=setosa,
                  kind='hex')

修改kind为'kde'来将直方图和散点图转换为核密度估计图，并将边际轴的留白大小设定为0：

ax = sns.jointplot(x='sepal_length',y='sepal_width',data=setosa,
                   kind='kde',
                   space=0,
                   color='g')

jointplot还支持图层叠加，如下面的例子，我们首先绘制出的联合图中kind限制为拟合线性回归直线，在此基础上利用.plot_joint方法叠加核密度估计图层：

ax = (sns.jointplot(x='sepal_length',y='sepal_width',data=setosa,
                   color='g',
                   kind='reg')).plot_joint(sns.kdeplot, zorder=0, n_levels=10)

根据你的具体需要还可以叠加出更加丰富的图像。

调大ratio参数使得边缘图更加小巧：

ax = sns.jointplot(x='sepal_length',y='sepal_width',data=setosa,
                   kind='kde',
                   space=0,
                   color='g',
                   ratio=15)

利用边缘图形参数字典为边缘图形添加rugplot的内容，并修改直方个数为15：

ax = sns.jointplot(x='sepal_length',y='sepal_width',data=setosa,
                   marginal_kws=dict(bins=15, rug=True),
                   linewidth=1,space=0)

实际上，如果你足够了解matplotlib与seaborn，可以通过各种组合得到信息量更丰富特别的图像！

以上就是本文的全部内容，如有笔误望指出！

上一篇下一篇

猜你喜欢

热点阅读