Python利用matplotlib实现制作动态条形图

制作思路

animation

大家好，本文将分享如何使用matplotlib制作动态条形图，制作的图很美，这个是我在之前发布的一篇中使用的图片，

效果如下

制作思路

为了方便大家学习，我将不直接进行讲解，而是以我是如何一步步制作的思路来介绍整个过程。

说到用 Python 制作动态图，首先想到的肯定是一些直接拿来就用的库，虽然我没做过，但是我相信一定有且不止一个，搜了一圈后发现有个bar chart race库看起来不错

毕竟有现成的轮子，只需要填充数据即可，但是研究了一番，正如我之前所说，越好看的图对数据的要求越高，这个动态图的库指定数据必须类似如下格式

即有时间和国家两个维度，通过时间的变化来动态变化数据，但是我的数据很简单，就一列人名一列数据如下

如果再将数据强行修改成该库所需要的格式，花费的功夫将更多！所以果断放弃，回到我熟悉的matplotlib上，首先很自然的想法就是，根据每一行，写一个循环，每次循环就创建一个条形图。

这个想法是完全可以的，也有很多人这么做了，但是有一个问题就是做出来的效果太难看了，并且每次循环都需要重新生成画布绘图需要一定时间，这也会让最后的动态图显的不连贯。所以稍作尝试后这个思路也被抛弃了。

animation

又经过一番探索后，终于找到一个合适的工具——animation模块，可以完美的实现我的需求，只需要将你的绘图函数写好，将涉及到的参数空出来，通过FuncAnimation来更新数据即可。

当然我也是第一次接触，虽了解思路但不清楚具体实现逻辑，所以先从官方拉一个demo来看看整个过程如何实现

fig, ax = plt.subplots()
xdata, ydata = [], []
ln, = ax.plot([], [], 'r-', animated=False)
def init():
    ax.set_xlim(0, 2*np.pi)
    ax.set_ylim(-1, 1)
    return ln,
def update(frame):
    xdata.append(frame)
    ydata.append(np.sin(frame))
    ln.set_data(xdata, ydata)
    return ln,
ani = FuncAnimation(fig, update, frames=np.linspace(0, 2*np.pi, 128),
                    init_func=init, blit=True)
HTML(ani.to_jshtml())

上面这段代码取自官方文档，并不难读懂，首先创建figure和axes，之后将涉及到的数据xdata，ydata空出来，再用update函数来更新数据并绘图，具体为将数据赋给frames，通过FuncAnimation实例传到update中实现迭代绘图，最后可以使用ipython的内置工具来将动态图展示在notebook中，效果如下