从零开始学matplotlib画图(三): 堆积图

51 阅读 0 评论 34 点赞

我是靠谱客的博主洁净天空，最近开发中收集的这篇文章主要介绍从零开始学matplotlib画图(三): 堆积图，觉得挺不错的，现在分享给大家，希望可以做个参考。

概述

堆积图常用于综合展示不同分类的指标趋势以及它们的总和的趋势。比如说，我们想看一下过去二十年来中国人口总量的变化趋势，同时，我们又想看一下男、女性人口各自的变化趋势，甚至我们还想看一下它们各自占比的变化趋势，这时，我们就可以用堆积图来更高效、更简洁地展示出来。

欢迎大家关注我的个人博客【数洞】【备用站】

我们举这样一个例子，有一个班里有20名学生，它们的编号分别是0-19，y1、y2、y3分别代表本次月考他们的语文、数学和英语的成绩，我们想观察这些学生的总成绩以及各科成绩的情况。

1. 堆积柱状图

首先，我们使用堆积柱状图来达到我们的目的：

import matplotlib.pyplot as plt
import numpy as np
# 生成数据
x = np.linspace(0, 20, 20)
y1 = np.random.randint(50, 100, 20)
y2 = np.random.randint(50, 100, 20)
y3 = np.random.randint(50, 100, 20)
# 堆积柱状图
plt.bar(x, y1, color='r', label='语文')
plt.bar(x, y2, bottom=y1, color='g', label='数学')
plt.bar(x, y3, bottom=y1+y2, color='c', label='英语')
# 显示范围
plt.xlim(-2, 22)
plt.ylim(0, 280)
# 添加图例
plt.legend(loc='upper right')
plt.grid(axis='y', color='gray', linestyle=':', linewidth=2)
plt.show()

可以看到，有四位同学的总成绩高于250分，还有些同学严重偏科，另外有一个同学平均下来在及格线下方……大家不同太为他担心，因为这些都是随机生成的数据。

我们还可以将柱子横过来：

import matplotlib.pyplot as plt
import numpy as np
# 生成数据
x = np.linspace(0, 20, 20)
y1 = np.random.randint(50, 100, 20)
y2 = np.random.randint(50, 100, 20)
y3 = np.random.randint(50, 100, 20)
# 堆积柱状图
plt.barh(x, y1, color='r', label='语文')
plt.barh(x, y2, left=y1, color='g', label='数学')
plt.barh(x, y3, left=y1+y2, color='c', label='英语')
# 显示范围
plt.ylim(-2, 22)
plt.xlim(0, 300)
# 添加图例
plt.legend(loc='lower right')
plt.show()

我们需要将bottom参数的名字改成left，这样更符合我们的视觉认知，其他参数上没有什么分别。

2. 堆积折线图

这次，我们假设小明参加了20次月考，我们想看下他的总成绩以及语数外三科成绩的变化趋势。

import matplotlib.pyplot as plt
import numpy as np
# 生成数据
# 假设一个班里有20名学生，x代表他们的编号，y1/y2/y3分别是他们本次月考的成绩。
x = np.linspace(0, 20, 20)
y1 = np.random.randint(50, 100, 20)
y2 = np.random.randint(50, 100, 20)
y3 = np.random.randint(50, 100, 20)
# 堆积柱状图
plt.stackplot(x, y1, y2, y3, baseline='zero', labels=['语文', '数学', '英语'], colors=['r', 'g', 'c'])
# 显示范围
plt.xlim(-2, 22)
plt.ylim(0, 300)
# 添加图例和网格线
plt.legend(loc='upper right')
plt.grid(axis='y', color='gray', linestyle=':', linewidth=2)
plt.show()

嗯……小明的成绩太不稳定了，不过还好，平均成绩还没到过及格线以下。从这张图里边，我们看不到明显的上升或者下降的趋势。

这样我们看到了成绩的变化趋势，但是我们不能清楚地分辨小明哪科成绩更好，哪科成绩是短板，因此我们需要看一下小明的总成绩中各科成绩占比的变化趋势。

import matplotlib.pyplot as plt
import numpy as np
# 生成数据
# 假设一个班里有20名学生，x代表他们的编号，y1/y2/y3分别是他们本次月考的成绩。
x = np.linspace(0, 20, 20)
y1 = np.random.randint(50, 100, 20)
y2 = np.random.randint(50, 100, 20)
y3 = np.random.randint(50, 100, 20)
# 计算百分比
y1p = y1 / (y1 + y2 + y3)
y2p = y2 / (y1 + y2 + y3)
y3p = y3 / (y1 + y2 + y3)
# 比例堆积柱状图
plt.stackplot(x, y1p, y2p, y3p, baseline='zero', labels=['语文', '数学', '英语'], colors=['r', 'g', 'c'])
# 显示范围
plt.xlim(0, 20)
plt.ylim(0, 1)
# 添加图例
plt.legend(loc='upper right')
plt.grid(axis='y', color='gray', linestyle=':', linewidth=2)
plt.show()