巧用 Matplotlib 动画,让你的 Python 可视化大放异彩

2019年1月5日 0条评论 67次阅读 0人点赞 9suan

作者：Tony Yiu

翻译：Hanz

整理：Lemonbit

译文出品：Python数据之道

— 1 —

前言

如果你对本文的代码感兴趣，可以去 Github （文末提供）里查看。第一次运行的时候会报一个错误（还没找到解决办法），不过只要再运行一次就正常了。

这篇文章虽然不是篇典型的数据科学类文章，不过它涉及到数据科学以及商业智能的应用。Python 的 Matplotlib 是最常用的图表绘制以及数据可视化库。我们对折线图、柱状图以及热力图都比较熟悉，但你知道用 Matplotlib 还能做简单的动画吗？

下面就是用 Matplotlib 制作动画的例子。展示的是 John Conway 的《The Game of Life》，这是一个 Metis（数据科学夏令营）中的编程挑战题目，同时给了我一个机会来制作我的第一个 Python 动画。看看结果的动图：

巧用 Matplotlib 动画,让你的 Python 可视化大放异彩插图(1)

这篇文章的重点还是主要放在 python 中如何用 Matploylib 制作动画。

但如果你不太熟悉模拟游戏的话（它更像是可以看的模拟动画，而非可以玩的游戏），我来给大家介绍一下规则：

一开始先设置一个 N×N 的网格（我的动画中用的是 50×50 ）；
接着随机地向格子中填充“小细胞”（一开始随机地从 2500 个格子中选取 1500 个进行填充）；
如果邻居小细胞少于等于 1 个，那格子中的小细胞会死掉；
如果邻居大于等于 4 个的也会死掉；
只有 2 个或 3 个邻居时可以生存；
空的格子中如果正好有 3 个邻居，则会长出 1 个新的“小细胞”；

— 2 —

建立网格

我们首先导入所需的库。




1
<span class="kwd" style="color: rgb(167, 29, 93);">import</span><span class="pln" style="color: rgb(22, 27, 29);"> time</span>




1
<span class="kwd" style="color: rgb(167, 29, 93);">from</span><span class="pln" style="color: rgb(22, 27, 29);"> </span><span class="typ" style="color: rgb(37, 127, 173);">IPython</span><span class="pln" style="color: rgb(22, 27, 29);"> </span><span class="kwd" style="color: rgb(167, 29, 93);">import</span><span class="pln" style="color: rgb(22, 27, 29);"> display</span>




1
<span class="kwd" style="color: rgb(167, 29, 93);">import</span><span class="pln" style="color: rgb(22, 27, 29);"> matplotlib.pyplot </span><span class="kwd" style="color: rgb(167, 29, 93);">as</span><span class="pln" style="color: rgb(22, 27, 29);"> plt</span>




1
<span class="kwd" style="color: rgb(167, 29, 93);">import</span><span class="pln" style="color: rgb(22, 27, 29);"> matplotlib.animation </span><span class="kwd" style="color: rgb(167, 29, 93);">as</span><span class="pln" style="color: rgb(22, 27, 29);"> animation</span>

我们会利用 Matploylib 动画模块中的

1
FuncAnimation()

函数。

1
FuncAnimation()

是通过多次调用一个函数并逐次更新图片来实现让图片动起来的。 我们来一步步地实现这个过程。

但首先，我们需要先初始化我们的网格。下面的几行代码用来存储我们输入的数据：

我们需要一个 50×50 大小的网格；
1
pad

变量可以使得计算邻居变得更容易。通过在边界外添加一层空白格子，我们就不需要额外再写一个逻辑来处理网格的边界。因此我们 50×50 的网格其实是被一圈空白格子包围着，这使得实际的 numpy 序列的大小为 52×52；
1
initial_cels

变量表示在网格启动的时候我们想要多少“小细胞”。他们会被随机地分布在网格上。




1
<span class="com" style="color: rgb(142, 142, 142);"># Input variables for the board</span>




1
<span class="pln" style="color: rgb(22, 27, 29);">boardsize = </span><span class="lit" style="color: rgb(147, 92, 37);">50</span><span class="pln" style="color: rgb(22, 27, 29);">        </span><span class="com" style="color: rgb(142, 142, 142);"># board will be X by X where X = boardsize</span>




1
<span class="pln" style="color: rgb(22, 27, 29);">pad = </span><span class="lit" style="color: rgb(147, 92, 37);">2</span><span class="pln" style="color: rgb(22, 27, 29);">               </span><span class="com" style="color: rgb(142, 142, 142);"># padded border, do not change this!</span>




1
<span class="pln" style="color: rgb(22, 27, 29);">initial_cells = </span><span class="lit" style="color: rgb(147, 92, 37);">1500</span><span class="pln" style="color: rgb(22, 27, 29);">  </span><span class="com" style="color: rgb(142, 142, 142);"># this number of initial cells will be placed</span>




1
<span class="pln" style="color: rgb(22, 27, 29);">                      </span><span class="com" style="color: rgb(142, 142, 142);"># in randomly generated positions</span>

接下来我们随机地生成一系列“小细胞”的初始坐标（上面我们选择了 1500 个）。把这些坐标存储在

<span class="pln" style="color: rgb(167, 29, 93);font-size: 14px;">pos_list</span>

变量中。




1
<span class="com" style="color: rgb(142, 142, 142);"># Get a list of random coordinates so that we can initialize</span>




1
<span class="com" style="color: rgb(142, 142, 142);"># board with randomly placed organisms</span>




1
<span class="pln" style="color: rgb(22, 27, 29);">pos_list = []</span>




1
<span class="kwd" style="color: rgb(167, 29, 93);">for</span><span class="pln" style="color: rgb(22, 27, 29);"> i </span><span class="kwd" style="color: rgb(167, 29, 93);">in</span><span class="pln" style="color: rgb(22, 27, 29);"> range(initial_cells):</span>




1
<span class="pln" style="color: rgb(22, 27, 29);">    pos_list.append([random.randint(</span><span class="lit" style="color: rgb(147, 92, 37);">1</span><span class="pun" style="color: rgb(22, 27, 29);">, boardsize),</span>




1
<span class="pln" style="color: rgb(22, 27, 29);">                     random.randint(</span><span class="lit" style="color: rgb(147, 92, 37);">1</span><span class="pun" style="color: rgb(22, 27, 29);">, boardsize)])</span>

然后我们是时候该初始化网格了。我们会用一组叫

<span class="pln" style="color: rgb(167, 29, 93);font-size: 14px;">my_board</span>

的 numpy 序列来代表我们的网格——我们先生成一个 52×52 数值为 0 的矩阵序列作为开始（比 50×50 大是由于增加了空白边缘），然后调用

<span class="pln" style="color: rgb(167, 29, 93);font-size: 14px;">init_board()</span>

函数来根据

<span class="pln" style="color: rgb(167, 29, 93);font-size: 14px;">pos_list</span>

中的坐标把“小细胞”填充到网格中。辅助函数的具体细节我不再展开讲了，不过我把他们都整理到我的 Github 上了。




1
<span class="com" style="color: rgb(142, 142, 142);"># Initialize the board</span>




1
<span class="pln" style="color: rgb(22, 27, 29);">my_board = np.zeros((boardsize+pad, boardsize+pad))</span>




1
<span class="pln" style="color: rgb(22, 27, 29);">my_board = init_board(pos_list, my_board)</span>

— 3 —

制作网格动画

这是我们最期待的部分——动画！首先，我们需要完善一些配置。下面的几行代码用来生成展示我们动画的

<span class="pln" style="color: rgb(167, 29, 93);font-size: 14px;">mtplotlib</span>

图框。




1
<span class="com" style="color: rgb(142, 142, 142);"># Required line for plotting the animation</span>




1
<span class="pun" style="color: rgb(22, 27, 29);">%matplotlib notebook</span>




1
<span class="com" style="color: rgb(142, 142, 142);"># Initialize the plot of the board that will be used for animation</span>




1
<span class="pln" style="color: rgb(22, 27, 29);">fig = plt.gcf()</span>

接下来制作我们的第一帧。

<span class="pln" style="color: rgb(167, 29, 93);font-size: 14px;">mtplotlib</span>

中的

<span class="pln" style="color: rgb(167, 29, 93);font-size: 14px;">imshow()</span>

函数可以接收一组

<span class="pln" style="color: rgb(167, 29, 93);font-size: 14px;">numpy</span>

矩阵然后返回一张图片。很酷吧！




1
<span class="com" style="color: rgb(142, 142, 142);"># Show first image - which is the initial board</span>




1
<span class="pln" style="color: rgb(22, 27, 29);">im = plt.imshow(my_board)</span>




1
<span class="pln" style="color: rgb(22, 27, 29);">plt.show()</span>

传入

<span class="pln" style="color: rgb(167, 29, 93);font-size: 14px;">imshow()</span>

的变量是我们的初始的网格

<span class="pln" style="color: rgb(167, 29, 93);font-size: 14px;">my_board</span>

。生成的图片长这样：

巧用 Matplotlib 动画,让你的 Python 可视化大放异彩插图(2)

现在我们需要写一个可以给

<span class="typ" style="color: rgb(167, 29, 93);font-size: 14px;">FuncAnimation()</span>

调用的辅助函数。

<span class="pln" style="color: rgb(167, 29, 93);font-size: 14px;">animate()</span>

函数接受一帧画面作为输入充当计数器。这个画面计数器就是

<span class="typ" style="color: rgb(167, 29, 93);font-size: 14px;">FuncAnimation()</span>

和

<span class="pln" style="color: rgb(167, 29, 93);font-size: 14px;">animate()</span>

函数沟通的桥梁——在每一个时间点（也就是每一帧），它都会调用一次

<span class="pln" style="color: rgb(167, 29, 93);font-size: 14px;">animate()</span>

。然后

<span class="pln" style="color: rgb(167, 29, 93);font-size: 14px;">animate()</span>

会逐次使用辅助函数

<span class="pln" style="color: rgb(167, 29, 93);font-size: 14px;">update_board()</span>

来对网格进行迭代。最后，

<span class="pln" style="color: rgb(167, 29, 93);font-size: 14px;">set_data()</span>

函数将图片更新为迭代后的网格，这就完成了。




1
<span class="com" style="color: rgb(142, 142, 142);"># Helper function that updates the board and returns a new image of</span>




1
<span class="com" style="color: rgb(142, 142, 142);"># the updated board animate is the function that FuncAnimation calls</span>




1
<span class="kwd" style="color: rgb(167, 29, 93);">def</span><span class="pln" style="color: rgb(22, 27, 29);"> animate(frame):</span>




1
<span class="pln" style="color: rgb(22, 27, 29);">    im.set_data(update_board(my_board))</span>




1
<span class="pln" style="color: rgb(22, 27, 29);">    </span><span class="kwd" style="color: rgb(167, 29, 93);">return</span><span class="pln" style="color: rgb(22, 27, 29);"> im,</span>

一切顺利！我们准备调用

<span class="typ" style="color: rgb(167, 29, 93);font-size: 14px;">FuncAnimation()</span>

函数了。注意输入的参数：

1
fig

是我们在前面创建的用来装载我们的动画的图形变量；
1
animate

是

1
FuncAnimation()

用画面计数器进行沟通的函数（自动传入，不需要特别声明）
1
frames

表示我们希望动画持续多少帧，在这里我们想要动画的长度为 200 帧；
1
interval

表示每一帧之间间隔的毫秒数。我们想要每帧之间间隔 50 毫秒。




1
<span class="com" style="color: rgb(142, 142, 142);"># This line creates the animation</span>




1
<span class="pln" style="color: rgb(22, 27, 29);">anim = animation.</span><span class="typ" style="color: rgb(37, 127, 173);">FuncAnimation</span><span class="pun" style="color: rgb(22, 27, 29);">(fig, animate, frames=</span><span class="lit" style="color: rgb(147, 92, 37);">200</span><span class="pun" style="color: rgb(22, 27, 29);">,</span>




1
<span class="pln" style="color: rgb(22, 27, 29);">                               interval=</span><span class="lit" style="color: rgb(147, 92, 37);">50</span><span class="pun" style="color: rgb(22, 27, 29);">)</span>

就这么简单！不是很难吧？为了庆祝我们成功制作动画，我再送大家一个动画：

巧用 Matplotlib 动画,让你的 Python 可视化大放异彩插图(3)

— 4 —

总结

希望这篇文章能帮到大家。在结束之前，让我来帮助大家脑补更多我们今天学到的动画功能在数据科学上的应用：

一个个地画出蒙特卡洛模拟数据，你能观察到最终的分布是如何逐步形成的；
按顺序遍历时间序列数据，可以描绘你的模型或数据在新的观察角度下有什么表现；
当你改变输入参数时，比如族群数，可以展现你的算法是如何划分族群的；
根据时间或不同的数据子集生成关联热力图，用于观察不同的样本是如何影响你的模型的预期参数的。

Github: https://github.com/yiuhyuk/game_of_life

来源：https://towardsdatascience.com/spice-up-your-python-visualizations-with-matplotlib-animations-d437d7e98e67

（完）

看完本文有收获？请转发分享给更多人

关注「Python那些事」，做全栈开发工程师

点「在看」的人都变好看了哦

九算AI实验室|18年专注网站优化-网站SEO-百度账户托管

巧用 Matplotlib 动画,让你的 Python 可视化大放异彩

朋友会在“发现-看一看”看到你“在看”的内容

朋友将在看一看看到

发布到看一看

本作品采用知识共享署名-相同方式共享 4.0 国际许可协议进行许可

九算AI实验室|18年专注网站优化-网站SEO-百度账户托管

朋友会在“发现-看一看”看到你“在看”的内容

朋友将在看一看看到

发布到看一看

本作品采用 知识共享署名-相同方式共享 4.0 国际许可协议 进行许可

本作品采用知识共享署名-相同方式共享 4.0 国际许可协议进行许可