照例引用官方介绍“Seaborn是一个带着定制主题和高级界面控制的Matplotlib扩展包,能让绘图变得更轻松”,这里我们只需要知道Seaborn是一个用于数据可视化的包就行了,可以让我们用最少的代码去绘制一些描述性统计的图,便于找寻各维度变量之间的特征和关系,需要快速了解Seaborn的,可以查看Seaborn官方教程。接下来我们将通过一个经典数据集——鸢尾花来初步认识Seaborn。
0.开始前的准备
Iris鸢尾花数据集是一个经典数据集,在统计学习和机器学习领域都经常被用作示例。数据集内包含3类共150条记录,每类各50个数据,每个数据都有4个属性:花萼长度、花萼宽度、花瓣长度、花瓣宽度,可以通过这4个属性预测鸢尾花卉属于(iris-setosa,iris-versicolour,iris-virginica)中的哪一类,但据说现实中,这三种花的基本判别依据其实是种子(因为花瓣非常容易枯萎)。
在分析这个数据集之前,让我们先做好相关准备,包括导入相关包、导入数据集等。
# 导入相关包
import numpy as np
import pandas as pd
import seaborn as sns
import matplotlib.pyplot as plt
# 导入数据集,也可以通过iris = sns.load_dataset("iris") 导入
iris = pd.read_csv('data/iris.csv', header=None)
#设置列名
iris.columns = ['sepal-length', 'sepal-width', 'petal-length', 'petal-width', 'species']
#查看数据集信息
iris.info()
<class 'pandas.core.frame.DataFrame'>
RangeIndex: 150 entries, 0 to 149
Data columns (total 5 columns):
sepal-length 150 non-null float64
sepal-width 150 non-null float64
petal-length 150 non-null float64
petal-width 150 non-null float64
species 150 non-null object
dtypes: float64(4), object(1)
memory usage: 5.9+ KB
# 查看数据集头5条记录
iris.head()
# 查看鸢尾花的所有种类
iris['species'].unique()
array(['Iris-setosa', 'Iris-versicolor', 'Iris-virginica'], dtype=object)
完成这些准备工作后,我们可以通过seaborn对该数据集进行可视化分析。
1.Stripplot & Swarmplot
# 设置seaborn样式
sns.set_style("whitegrid")
seaborn.stripplot(x=None, y=None, hue=None, data=None, order=None, hue_order=None, jitter=False, dodge=False, orient=None, color=None, palette=None, size=5, edgecolor='gray', linewidth=0, ax=None, **kwargs)
Stripplot主要用于绘制散点图,下面将鸢尾花数据集中的花萼长度属性按品种分类绘制成散点图分析:
ax = sns.stripplot(x="species", y="sepal-length", data=iris)
通过jitter
属性为散点图设置抖动,以便更好地观察数据。
ax = sns.stripplot(x="species", y="sepal-length", data=iris, jitter=True)
ax = sns.stripplot(x="species", y="sepal-length", data=iris, jitter=0.05)
改变轴向观察数据。
ax = sns.stripplot(x="sepal-length", y="species", data=iris, jitter=0.05)
通过改变颜色、样式、大小等更好地观察数据。
antV = ['#1890FF', '#2FC25B', '#FACC14', '#223273', '#8543E0', '#13C2C2', '#3436c7', '#F04864']
ax = sns.stripplot(x="sepal-length", y="species", data=iris, jitter=0.05, palette=antV)
ax = sns.stripplot("species", "sepal-length", data=iris, palette="Set2", size=12, marker="D",
jitter=0.05, edgecolor="gray", alpha=.25)
seaborn.swarmplot(x=None, y=None, hue=None, data=None, order=None, hue_order=None, dodge=False, orient=None, color=None, palette=None, size=5, edgecolor='gray', linewidth=0, ax=None, **kwargs)
Swarmplot也是绘制散点图,但它会通过算法,在类别坐标轴的方向上延展那些原本重合的点,与通过jitter
属性增加抖动有异曲同工之妙。
ax = sns.swarmplot(x="species", y="sepal-length", data=iris)
现在我们通过Swarmplot观察花萼长度、花萼宽度、花瓣长度、花瓣宽度与类别之间的关系。
f, axes = plt.subplots(2, 2, figsize=(12, 12), sharex=True)
sns.despine(left=True)
sns.swarmplot(x="species", y="sepal-length", data=iris, palette=antV, ax=axes[0, 0])
sns.swarmplot(x="species", y="sepal-width", data=iris, palette=antV, ax=axes[0, 1])
sns.swarmplot(x="species", y="petal-length", data=iris, palette=antV, ax=axes[1, 0])
sns.swarmplot(x="species", y="petal-width", data=iris, palette=antV, ax=axes[1, 1])
plt.setp(axes, yticks=[])
plt.tight_layout()
2.Boxplot & Violinplot
seaborn.boxplot(x=None, y=None, hue=None, data=None, order=None, hue_order=None, orient=None, color=None, palette=None,
saturation=0.75, width=0.8, dodge=True, fliersize=5, linewidth=None, whis=1.5, notch=False, ax=None, **kwargs)
通过Boxplot可以看到数据的最大值、上四分位数Q3、中位数、下四分位数Q1、最小值和异常值的分布。
f, axes = plt.subplots(2, 2, figsize=(12, 12), sharex=True)
sns.despine(left=True)
sns.boxplot(x="species", y="sepal-length", data=iris, palette='Set2', ax=axes[0, 0])
sns.boxplot(x="species", y="sepal-width", data=iris, palette='Set2', ax=axes[0, 1])
sns.boxplot(x="species", y="petal-length", data=iris, palette='Set2', ax=axes[1, 0])
sns.boxplot(x="species", y="petal-width", data=iris, palette='Set2', ax=axes[1, 1])
plt.setp(axes, yticks=[])
plt.tight_layout()
seaborn.violinplot(x=None, y=None, hue=None, data=None, order=None, hue_order=None, bw='scott', cut=2, scale='area', scale_hue=True, gridsize=100, width=0.8, inner='box', split=False, dodge=True, orient=None, linewidth=None, color=None, palette=None, saturation=0.75, ax=None, **kwargs)
Violinplot与Boxplot相似,但其图形如同小提琴般,可以更好地展现出数据的量化形态。
# Set up the matplotlib figure
f, axes = plt.subplots(2, 2, figsize=(12, 12), sharex=True)
sns.despine(left=True)
sns.violinplot(x="species", y="sepal-length", data=iris, palette='Set2', ax=axes[0, 0])
sns.violinplot(x="species", y="sepal-width", data=iris, palette='Set2', ax=axes[0, 1])
sns.violinplot(x="species", y="petal-length", data=iris, palette='Set2', ax=axes[1, 0])
sns.violinplot(x="species", y="petal-width", data=iris, palette='Set2', ax=axes[1, 1])
plt.setp(axes, yticks=[])
plt.tight_layout()
可以Stripplot & Swarmplot与Boxplot & Violinplot将相组合,以更好地查看数据分布情况。
ax = sns.violinplot(x="species", y="sepal-length", data=iris, palette='Set2')
ax = sns.swarmplot(x="species", y="sepal-length", data=iris, color="white", edgecolor="gray")
3.Barplot
seaborn.barplot(x=None, y=None, hue=None, data=None, order=None, hue_order=None, estimator=<function mean>, ci=95, n_boot=1000, units=None, orient=None, color=None, palette=None, saturation=0.75, errcolor='.26', errwidth=None, capsize=None, dodge=True, ax=None, **kwargs)
Barplot主要是绘制条形图,利用条形的高度反映数值变量的集中趋势,在条形顶部还有误差棒。
f, axes = plt.subplots(2, 2, figsize=(10, 10), sharex=True)
sns.despine(left=True)
sns.barplot(x="species", y="sepal-length", data=iris, palette='Set2', ax=axes[0, 0])
sns.barplot(x="species", y="sepal-width", data=iris, palette='Set2', ax=axes[0, 1])
sns.barplot(x="species", y="petal-length", data=iris, palette='Set2', ax=axes[1, 0])
sns.barplot(x="species", y="petal-width", data=iris, palette='Set2', ax=axes[1, 1])
plt.setp(axes, yticks=[])
plt.tight_layout()
4.Countplot
seaborn.countplot(x=None, y=None, hue=None, data=None, order=None, hue_order=None, orient=None, color=None, palette=None, saturation=0.75, dodge=True, ax=None, **kwargs)
Countplot可以直观地查看每个类别下有多少个观察值。
ax = sns.countplot(y="species", data=iris, palette="Set3")
5.Pointplot
seaborn.pointplot(x=None, y=None, hue=None, data=None, order=None, hue_order=None, estimator=<function mean>, ci=95, n_boot=1000, units=None, markers='o', linestyles='-', dodge=False, join=True, scale=1, orient=None, color=None, palette=None, errwidth=None, capsize=None, ax=None, **kwargs)
Pointplot可以通过斜率很容易地看出各类别间的主要关系。
f, axes = plt.subplots(2, 2, figsize=(10, 10), sharex=True)
sns.despine(left=True)
sns.pointplot(x="species", y="sepal-length", data=iris, ax=axes[0, 0])
sns.pointplot(x="species", y="sepal-width", data=iris, ax=axes[0, 1])
sns.pointplot(x="species", y="petal-length", data=iris, ax=axes[1, 0])
sns.pointplot(x="species", y="petal-width", data=iris, ax=axes[1, 1])
plt.setp(axes, yticks=[])
plt.tight_layout()
6. Factorplot
seaborn.factorplot(x=None, y=None, hue=None, data=None, row=None, col=None, col_wrap=None, estimator=<function mean>, ci=95, n_boot=1000, units=None, order=None, hue_order=None, row_order=None, col_order=None, kind='point', size=4, aspect=1, orient=None, color=None, palette=None, legend=True, legend_out=True, sharex=True, sharey=True, margin_titles=False, facet_kws=None, **kwargs)
Factorplot将上述的函数与FacetGrid
结合起来,通过设置kind
属性,可以变化成不同的图形。
sns.factorplot(x="species", y="sepal-length", data=iris, kind="violin")
sns.factorplot(x="species", y="sepal-width", data=iris, kind="box");
7. PairGrid
class seaborn.PairGrid(data, hue=None, hue_order=None, palette=None, hue_kws=None, vars=None, x_vars=None, y_vars=None, diag_sharey=True, size=2.5, aspect=1, despine=True, dropna=True)
PairGrid可以用于绘制展现数据集内多个变量之间关系的矩阵图。
g = sns.PairGrid(data=iris, palette=antV, hue="species")
g = g.map(plt.scatter)
g = sns.PairGrid(data=iris, palette=antV, hue="species")
g = g.map_diag(plt.hist)
g = g.map_offdiag(plt.scatter)
g = g.add_legend()