python用matplotlib绘制并列双柱状图并标注数值
Python项目需要画两组数据的双柱状图,以下以一周七天两位小朋友吃糖颗数为例进行演示,用matplotlib库实现
·
项目场景:
Python项目需要画两组数据的双柱状图,以下以一周七天两位小朋友吃糖颗数为例进行演示,用matplotlib库实现
代码:
import matplotlib
import matplotlib.pyplot as plt
import numpy as np
def drawHistogram():
matplotlib.rc("font", family='MicroSoft YaHei')
list1 = np.array([5, 2, 1, 0, 8, 0, 6]) # 柱状图第一组数据
list2 = np.array([9, 5, 1, 2, 9, 2, 0]) # 柱状图第二组数据
length = len(list1)
x = np.arange(length) # 横坐标范围
listDate = ["星期一", "星期二", "星期三", "星期四", "星期五", "星期六", "星期天"]
plt.figure()
total_width, n = 0.8, 2 # 柱状图总宽度,有几组数据
width = total_width / n # 单个柱状图的宽度
x1 = x - width / 2 # 第一组数据柱状图横坐标起始位置
x2 = x1 + width # 第二组数据柱状图横坐标起始位置
plt.title("一周每天吃悠哈软糖颗数柱状图") # 柱状图标题
# plt.xlabel("星期") # 横坐标label 此处可以不添加
plt.ylabel("吃悠哈软糖颗数(个)") # 纵坐标label
plt.bar(x1, list1, width=width, label="小s吃糖数")
plt.bar(x2, list2, width=width, label="小y吃糖数")
plt.xticks(x, listDate) # 用星期几替换横坐标x的值
plt.legend() # 给出图例
plt.show()
if __name__ == '__main__':
drawHistogram()
效果图:
扩展功能及代码:
扩展功能一
如果横坐标标签比较长或是文字比较多,以一定角度倾斜展示,上文中代码这一行:
plt.xticks(x, listDate)
可以改为:
plt.xticks(x, listDate, rotation=30) # rotation为标签旋转角度
横坐标标签旋转30°效果如下:
横坐标标签旋转90°效果如下:
扩展功能二
如果希望具体的数据值展示在柱状图中,可以在代码 plt.legend()
前加入如下代码:
for a, b in zip(x1, list1):
plt.text(a, b + 0.1, '%.0f' % b, ha='center', va='bottom', fontsize=7)
for a, b in zip(x2, list2):
plt.text(a, b + 0.1, '%.0f' % b, ha='center', va='bottom', fontsize=7)
加了具体数值的柱状图效果如下:
后续有时间再继续补充扩展功能哦~
更多推荐
已为社区贡献16条内容
所有评论(0)