在matplotlib
的基础上,扩展出了Seaborn
,Seaborn
是对matplotlib
的api进一步的封装,在已有的基础上使得一些图形更容易绘制,在这个介绍几个Seaborn
扩展出的特性。官网在 http://seaborn.pydata.org/index.html) ,里面有更多的扩展可以供进一步的参看。
分类显示
在机器学习上最经常碰到的是如何把两类型的数据进行区分,在图标上最容易理解的是把不同类型的数据按不同的颜色进行显示。
seaborn
自带一些实验的数据,如’iris’,他包含如下的属性:
sepal_length
sepal_width
petal_length
petal_width
species
。 其中species
表示物种的类型,有这些物种setosa
, virginica
, versicolor
这三种物种。如下需要按照物种标记不同的颜色。
import seaborn as sns
import matplotlib.pyplot as plt
df = sns.load_dataset('iris')
sns.lmplot(x="sepal_length", y="sepal_width", data=df, fit_reg=False, hue='species', legend=False)
plt.legend(loc='lower right')
plt.show()
不同的分类显示: