前述学习过程中,已经知晓普通直方图绘制和堆叠直方图绘制,参考链接如下:
西猫雷婶-CSDN博客
有时候,我们还会遇到并列直方图绘制的需求,今天就探索一下。
按照惯例,我们先来到官网:
https://matplotlib.org/stable/gallery/lines_bars_and_markers/barchart.html#sphx-glr-gallery-lines-bars-and-markers-barchart-py
图1
找到grouped bar chart with labels后。直奔示例。
然后进入学习模式。
官网代码前两行依旧岿然不动,引入计算模块numpy用于数值运算,画图模块matplotlib用于绘图。
然后定义了类别和对应的数据:
之后定义了要画图,使用ax控制坐标轴属性:
之后开始画直方图,用for循环实现方块输出,调整multiplier实现方块位移:
最后设置坐标轴属性,输出图形:
ax.set_xticks允许一个类别多个数据,这是并列直方图出现的直接推手。
图2
尝试把padding从3改为50后,各类被数据都往上漂移了:
图3
multiplier控制各方块的间隙,尝试把multiplier在for循环中的增量减小到0.5:
图4
由图4可见,方块大小不再一致。
实际上这是因为:新的方块只移动了一半(multiplier=1),所以被旧方块盖住了,最佳的移动还是要按照multiplier=1,让整个方块都显示出来,这样才会让图片更好看。
要想把并列方块数量增多,因修改penguin_means中的一维数组列数,每一列方块要使用不同的颜色区分,所以ncols也要跟随变化。某一类别下有几个并列方块,就有几种颜色。
改后:
运行后:
图5
我们很难过的发现所有方块挤在了一起,所以我们要去需改方块宽度:由0.25改为0.1
运行后:
图6
可见,图形没有挤在一起,出现了6个颜色图例。
本次改写完整代码如下:
# data from https://allisonhorst.github.io/palmerpenguins/
import matplotlib.pyplot as plt
import numpy as np
species = ("Adelie", "Chinstrap", "Gentoo")#类别,3种
penguin_means = {#这里实际上给每种类别定义了3个数据
'Bill Depth': (18.35, 18.43, 14.98),
'Bill Length': (38.79, 48.83, 47.50),
'Flipper Length': (189.95, 195.82, 217.19),
'add Bill Depth': (18.35, 18.43, 14.98),
'add Bill Length': (38.79, 48.83, 47.50),
'add Flipper Length': (189.95, 195.82, 217.19),
}
x = np.arange(len(species)) # the label locations,按照顺序排列species
width = 0.1 # the width of the bars 直方图方块宽度
multiplier = 0 #定义乘法因子
fig, ax = plt.subplots(layout='constrained')#定义一个带坐标轴的图,坐标轴属性用ax控制
for attribute, measurement in penguin_means.items():#for函数用于遍历penguin_means内部数组
offset = width * multiplier #直方图间距偏移量定义
rects = ax.bar(x + offset, measurement, width, label=attribute)
# 画直方图,x+offset代表(类别的位置+偏移量),偏移之后才能并列放置,都则就会重叠;
# measurement指将penguin_means中的数据按顺序赋值给x+offset对应的类别;
# width为方块宽度
# label就是将penguin_means中的数据类别作为标签
ax.bar_label(rects, padding=3) #输出标签,padding指和方块顶部的距离,一个三个类别,所以设定值为3
multiplier += 1 #乘法因子扩大,带来每个类别内部方块的移动
# Add some text for labels, title and custom x-axis tick labels, etc.
ax.set_ylabel('Length (mm)') #设定Y轴标签
ax.set_title('Penguin attributes by species') #设定图名
ax.set_xticks(x + width, species) #设定x轴,允许放多个类别
ax.legend(loc='upper left', ncols=6) #设定图例,位置为左上upper left,图例的颜色数量是ncols=3
ax.set_ylim(0, 250) #设定y轴范围是(0,250)
plt.show() #输出图形
本次学习探讨了并列直方图的for循环画法,并学会了修改各类别标签的位置、各类别方块的移动和多个方块的设定。