有时候在可视化数据时,为了能够在一幅图上更多的展现信息,会使用到不同类型的图结合。
本篇主要讲解如何结合使用常见的柱状图和线图,让你的绘图不再单调。
主要的思路是在一个figure上,先画柱图再用线将柱子连接起来,形成连贯的感觉。
具体例子代码如下:
y1 = [10,20,24,42]
y2 = [71,70,74,58]
y3 = [19,10,2,0]
texts = ['Harm', 'Neutral', 'Benefit']
colors = ['#ca4235','grey', '#3f834a']
ig, ax=plt.subplots(figsize=(6,5))
plt.bar(range(1,5), y1, align='edge', label='Harm', color=['#ca4235'], width=0.4)
plt.bar(range(1,5), y2, align='edge', bottom=y1,label='Neutral',color=['#aaa9a9'], width =0.4)
for i in range(0,4):
y2[i] += y1[i]
plt.bar(range(1,5), y3, align='edge', bottom=y2,label='Benefit', color=['#3f834a'], width =0.4)
for i in range(1,4):
if y1[i-1]!=0 and y1[i]!=0:
plt.plot([i+0.4,i+1], y1[i-1:i+1], color='#ca4235')
if y3[i-1]!=0 and y3[i]!=0:
plt.plot([i+0.4,i+1], y2[i-1:i+1], color='#3f834a')
location = [0, np.min(y2), 98]
for i in range(3):
plt.text(4.6, location[i], texts[i], color=colors[i], fontsize=16, weight='550')
ax.spines['top'].set_visible(False)
ax.spines['right'].set_visible(False)
ax.spines['bottom'].set_visible(False)
ax.set_xticks([])
plt.ylabel("Trial Conclusion for Stroke (%)", fontsize=15)
plt.show()
这里需要有几点要注意:
1. 对于连线时,为了统一都将线从柱子右侧出,连到下一个柱子左侧,这里是分别对每个线单独画,设定固定的坐标。如果有其他更好的办法,可以留言。
2. 隐藏x轴的刻度坐标 使用set_xticks(),参数为空列表。
3. 对于值为0的时候,不画线。这里需要在画线图的时候,对值做判断。
可视化结果如下所示:
如果本篇讲解的内容,对你有帮助,欢迎点赞留言收藏,给予支持!