Matplotlib条形图之分组条形图和堆叠条形图详解
分组条形图
拓展一下问题复杂度:使用水平条形图展示每位员工前三个月的销售额。此时,我们需要将每位员工的销售额按月分组,分别绘制条形图进行展示。
import numpy as np import matplotlib.pyplot as plt plt.rcParams['font.sans-serif'] = ['SimHei'] # 用来正常显示中文标签 plt.rcParams['axes.unicode_minus'] = False # 用来正常显示负号 member = [u'小红', u'小王', u'小李', u'小张'] sales_jan = [30, 42, 25, 35] # 一月销售额 sales_feb = [60, 55, 10, 27] # 二月销售额 sales_mar = [40, 20, 5, 68] # 三月销售额 bar_width = 0.2 # 设置分组条形的宽度 # 绘图 plt.bar(range(4), sales_jan, label=u'一月', width=bar_width, align='center', color='steelblue', alpha=0.7) # 也可以使用numpy模块的arange()函数构造横坐标 plt.bar(np.arange(4) + bar_width, sales_feb, label=u'二月', color='indianred', alpha=0.7, width=bar_width) plt.bar(np.arange(4) + bar_width * 2, sales_mar, label=u'三月', color='green', alpha=0.7, width=bar_width) # 添加Y轴标签 plt.ylabel(u'月度销售额(万元)') # 添加标题 plt.title(u'员工第一季度销售额对比') # 添加刻度标签 plt.xticks(np.arange(4) + bar_width, member) # 添加图例 plt.legend() plt.xlim(-0.5, 4.5) plt.show()
分组条形图比简单条形图的复杂之处在于,在放数据 x
轴坐标和刻度位置时,需要进行额外的计算和调整。在执行坐标的计算时,推荐使用 numpy ,因为它支持广播机制,向量化的算术运算更加简单。例如,上例中在设置第二个和第三个条形的 x
轴坐标时,我们使用了 np.arange(4) + bar_width
和 np.arange(4) + bar_width * 2
,而利用 Python 列表实现将十分复杂。
堆叠条形图
堆叠条形图是分组条形图展示的另一种形式,它把分类的数据堆叠在一起,显得更简约紧密,同时提供了求和信息。在实现上,绘制的思路与条形图相似,不过前者是垂直偏移,后者是水平偏移。
import numpy as np import matplotlib.pyplot as plt plt.rcParams['font.sans-serif'] = ['SimHei'] # 用来正常显示中文标签 plt.rcParams['axes.unicode_minus'] = False # 用来正常显示负号 member = [u'小红', u'小王', u'小李', u'小张'] sales_jan = [30, 42, 25, 35] # 一月销售额 sales_feb = [60, 55, 10, 27] # 二月销售额 sales_mar = [40, 20, 5, 68] # 三月销售额 bar_width = 0.2 # 设置分组条形的宽度 # 绘图 plt.bar(np.arange(4), sales_jan, label=u'一月', color='steelblue', alpha=0.7) plt.bar(np.arange(4), sales_feb, bottom=sales_jan, label=u'二月', color='indianred', alpha=0.7) plt.bar(np.arange(4), sales_mar, bottom=np.array(sales_jan) + np.array(sales_feb), label=u'三月', color='green', alpha=0.7) # 添加Y轴标签 plt.ylabel(u'月度销售额(万元)') # 添加标题 plt.title(u'员工第一季度销售额对比') # 添加刻度标签 plt.xticks(np.arange(4), member) # 添加图例 plt.legend() plt.xlim(-0.5, 4.5) plt.show()
这里有两点非常关键:一是 botom
选项的使用让数据在该基础之上有一个偏移;二是 NumPy 模块 array()
函数的使用,将列表类型的数据转换为 ndarray
,以便元素级别(向量化)运算。
总结
本篇文章就到这里了,希望能够给你带来帮助,也希望您能够多多关注脚本之家的更多内容!
相关文章
Django静态文件配置request对象方法ORM操作讲解
这篇文章主要为大家介绍了Django静态文件配置request对象方法ORM操作,有需要的朋友可以借鉴参考下,希望能够有所帮助,祝大家多多进步,早日升职加薪2022-09-09python3.6+django2.0+mysql搭建网站过程详解
这篇文章主要介绍了python3.6+django2.0+mysql搭建网站过程详解,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下2019-07-07深入理解Python虚拟机中列表(list)的实现原理及源码剖析
在本篇文章当中主要给大家介绍 cpython 虚拟机当中针对列表的实现,在 Python 中,List 是一种非常常用的数据类型,可以存储任何类型的数据,并且支持各种操作,如添加、删除、查找、切片等,在本篇文章当中将深入去分析这一点是如何实现的2023-03-03flask操作数据库插件Flask-SQLAlchemy的使用
Python 中最广泛使用的ORM框架是SQLAlchemy,它是一个很强大的关系型数据库框架,本文就来介绍一下flask操作数据库插件Flask-SQLAlchemy的使用,感兴趣的可以了解一下2023-09-09
最新评论