公众号后台回复“图书“,了解更多号主新书内容
作者:宁海涛
来源:DataCharm
上期推文推出第一篇基础图表绘制-R-ggplot2 基础图表绘制-散点图 的绘制推文,得到了很多小伙伴的喜欢,也是我更加想使这个系列做的更加完善和系统,我之前也有说过,会推出Python和R的两个版本绘制教程,接下来我们就推出基础散点图的Python绘制版本。本期主要涉及的知识点如下:
Python-seaborn 绘制多类别散点图
seaborn 定制化美化设置
Python-seaborn 绘制多类别散点图
由于涉及的图表类型为多类别散点图的绘制,在使用常规matplotlib进行绘制时会显得格外繁琐,所以我们选择了对matplotlib进行了更高级的API封装,使作图更加容易的seaborn包进行图表的绘制,更多seaborn 介绍,大家可以直接去seaborn官网进行相关资料的查阅。数据的读取使用的功能强大的数据处理包 pandas ,这里只是进行简单的删除空值操作,直接使用dropna() 函数操作即可,我们直接预览数据,如下(部分):由于我们直接使用了seaborn进行图表的绘制,绘图代码也得到了极大的简化,默认的绘图代码如下:
fig, ax = plt.subplots(figsize=(6,5),dpi=200)
scatter = sns.scatterplot(data=penguins_df,x="bill_length_mm",y="bill_depth_mm",hue="species",
size="body_mass_g",ec="k",alpha=.9,ax=ax)
scatter.legend()
ax.text(.91,-.1,'\nVisualization by DataCharm',transform = ax.transAxes,
ha='center', va='center',fontsize = 7,color='black')
可视化效果如下:
这配色、这布局······,这个图表中主要存在如下问题:
刻度、轴脊、XY刻度标签等属性过于简陋。
图例添加太随意,需要定制化操作(重点)
整体的绘图主题过于简单。
针对以上几个问题,接下来我们对其进行美化操作。
seaborn 定制化美化操作
详细的美化操作对于seaborn来说,代码过多,且需记住的绘图函数也较多,这里和R-ggplot2 绘图一样,我们直接选择matplotlib 绘图主题进行设置即可,此外,我们还对图例等图元素进行设置,具体代码如下:
plt.rcParams["font.family"] = "Arial-Narrow"
palette = ("#FF8C00", "#A034F0", "#159090")
plt.style.use('fivethirtyeight')
fig, ax = plt.subplots(figsize=(6,5),dpi=200)
scatter = sns.scatterplot(data=penguins_df,x="bill_length_mm",y="bill_depth_mm",hue="species",size="body_mass_g",
palette=palette,ec="white",alpha=.9,ax=ax)
#自定义图例设置
handles,labels = scatter.get_legend_handles_labels()
labels_size = labels[5:]
labels_size = [str(int(i)/1000)+" kg" for i in labels_size]
scatter.legend(handles=handles[1:4] + handles[5:], labels=labels[1:4] + labels_size ,ncol=1,loc="lower right",
frameon=False,fontsize=7,markerscale=1,bbox_to_anchor=(1, 0.0))
scatter.set_xlabel("Bill length (mm)")
scatter.set_ylabel("Bill depth (mm)")
#添加标题
ax.text(-.08,1.1,"Base Charts in Python Exercise 01: Point Charts",
transform = ax.transAxes,color='k',ha='left',va='center',size=15,fontweight='extra bold')
ax.text(.91,-.1,'\nVisualization by DataCharm',transform = ax.transAxes,
ha='center', va='center',fontsize = 7,color='black')
plt.show()
这里注意以下几点:
绘图主题的选择
plt.style.use('fivethirtyeight')
这样就选择了对应的绘图主题。
自定义图例设置
handles,labels = scatter.get_legend_handles_labels()
labels_size = labels[5:]
labels_size = [str(int(i)/1000)+" kg" for i in labels_size]
scatter.legend(handles=handles[1:4] + handles[5:], labels=labels[1:4] + labels_size ,ncol=1,loc="lower right",
frameon=False,fontsize=7,markerscale=1,bbox_to_anchor=(1, 0.0))
该部分代码就实现了图例的定制化操作,具体包括:图例元素选择、位置以及排列方式等.
这里指出一下:由于seaborn是对matplotlib的高度封装,这也导致其对个别图表元素的定制化设置就相对较难(如图例)
最终的可视化效果如下:
当然我们还可以将图例放置在上方并排显示,对应修改代码如下:
scatter.legend(handles=handles[1:4] + handles[5:], labels=labels[1:4] + labels_size ,ncol=8,loc="upper center",
frameon=False,fontsize=7,markerscale=1,bbox_to_anchor=(0.5, 1.03))
可视化效果如下:
总结
本期推文我们推出了基础散点图的Python绘制版本,希望可以满足喜欢使用Python绘图的小伙伴。大家有啥意见也可以在文末 读者讨论 区进行谈论交流啊。
◆ ◆ ◆ ◆ ◆
麟哥新书已经在京东上架了,我写了本书:《拿下Offer-数据分析师求职面试指南》,目前京东正在举行100-40活动,大家可以用相当于原价5折的预购价格购买,还是非常划算的:
数据森麟公众号的交流群已经建立,许多小伙伴已经加入其中,感谢大家的支持。大家可以在群里交流关于数据分析&数据挖掘的相关内容,还没有加入的小伙伴可以扫描下方管理员二维码,进群前一定要关注公众号奥,关注后让管理员帮忙拉进群,期待大家的加入。
管理员二维码:
猜你喜欢
● 卧槽!原来爬取B站弹幕这么简单● 厉害了!麟哥新书登顶京东销量排行榜!● 笑死人不偿命的知乎沙雕问题排行榜
● 用Python扒出B站那些“惊为天人”的阿婆主!● 你相信逛B站也能学编程吗