学习
实践
活动
工具
TVP
写文章

使用Python生成动态图表,一个很Cool的教程

作者 | Viviane

译者 | 刘旭坤

编辑 | Jane

出品 | AI科技大本营

【导读】Matplotlib 和 Seaborn 用来做数据固定的静态图表很不错,但如果需要动态显示数据的变化过程用这俩库就有点难度了。如果能用短视频来动态展示图表中数据的变化是不是就有点厉害了?而且不用学新的库,就用Matplotlib 和 Seaborn,是不是更厉害了!今天就教大家一个很酷的教程,马上动起来。

下载地址:

https://www.drugabuse.gov/sites/default/files/overdose_data_1999-2015.xls

除了用 Matplotlib 和 Seaborn 来作图我们还用到了 Numpy 和 Pandas 来处理数据。先把需要的库都 import 进来:

接下来把数据从下载的Excel文件读进来,我们用Pandas来读所以读进来就是一个DataFrame。这里我们还写了一个传入行数只读取一行数据的函数。这是为了以后给不同药物分别做图的时候读数据方便一些。

这些准备工作做好我们就可以开始作图了。如果你用 Jupyter Notebook的 话记得加图表页内显示的命令%matplotlib notebook。

我们先用刚才写的get_data函数把因为海洛因死亡的人数读出来。这个 DataFrame 有两列,分别是年份和死亡人数。

接下来我们初始化一个 ffmpeg 输出流。这里我设置帧率 20 码率 1800 ,当然你自己可以改帧率和码率。

下面我们需要创建图表和横纵坐标。这里要把数据范围定死不然数据更新的时候 Matplotlib 会自动更新数据范围我们的动图数据范围就会来回变。

绘图中最重要的就是下面这个 animate 函数,它的参数i指的是帧数。我们通过参数i来选择这一帧应该显示的数据然后用 Seaborn 来画一个折线图。最后两行改改字体和折线的宽度让图好看一点。

要让图表动起来我们得把刚才定义的 animate 函数传给matplotlib.animation.FuncAnimation。除了animate,FuncAnimation还有一个参数frames,这个参数的意思是说我们这段动画想一共要多少帧。这里 frames 的值是 17 帧,所以animate函数会被调用17次。

最后我们把这段动画存成 mp4 格式就行了。如果想先看看效果,可以用plt.show()。

最后出来的效果是这样的:

意思是有了但给人感觉数据的跳跃有点太快了,所以我们可能得给数据点中间插点值。插值可以用下面的augment函数:

用augment函数处理完数据我们还要调整一下 FuncAnimation 函数中的帧数。这里我给 augment 传入的numsteps是 10,也就是说 augment 后从99 年到 15 年的 16 个数据点变成了 160 个,所以帧数也要设成 160。调整完的图看着顺溜多了,不过在数据增减变化的地方还是能看到很硬的拐弯。

为了让这些拐角也平滑一点我们参考了下面这个链接里的高斯平滑算法。这个链接里也介绍了其他的平滑算法。

https://www.swharden.com/wp/2008-11-17-linear-data-smoothing-in-python/

此外我们还可以给图片加点背景色。

大功告成!其实不是大功,只是用 Matplotlib 制作动态图表的一个很基本的例子。不过原理都是一样的,在 animate 函数里绘图然后调整调整合适的参数什么图都能动起来。最后希望大家顺利操作起来!

https://towardsdatascienc e.com/how-to-create-animated-graphs-in-python-bb619cc2dec1

--【完】--

  • 发表于:
  • 原文链接https://kuaibao.qq.com/s/20180923A0V3HT00?refer=cp_1026
  • 腾讯「腾讯云开发者社区」是腾讯内容开放平台帐号(企鹅号)传播渠道之一,根据《腾讯内容开放平台服务协议》转载发布内容。
  • 如有侵权,请联系 cloudcommunity@tencent.com 删除。

扫码关注腾讯云开发者

领取腾讯云代金券