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

2024-01-20 07:10

本文主要是介绍巧用 Matplotlib 动画,让你的 Python 可视化大放异彩,希望对大家解决编程问题提供一定的参考价值,需要的开发者们随着小编来一起学习吧!

640?wx_fmt=jpeg

柠檬2069
点击右侧关注,遇见更好的自己!

巧用 Matplotlib 动画

让你的 Python 可视化大放异彩

640?wx_fmt=jpeg

作者:Tony Yiu

翻译:Hanz

整理:Lemonbit

译文出品:Python数据之道

—  1 

前言

如果你对本文的代码感兴趣,可以去 Github (文末提供)里查看。第一次运行的时候会报一个错误(还没找到解决办法),不过只要再运行一次就正常了。
这篇文章虽然不是篇典型的数据科学类文章,不过它涉及到数据科学以及商业智能的应用。Python 的 Matplotlib 是最常用的图表绘制以及数据可视化库。我们对折线图、柱状图以及热力图都比较熟悉,但你知道用 Matplotlib 还能做简单的动画吗?
下面就是用 Matplotlib 制作动画的例子。展示的是 John Conway 的 《The Game of Life》,这是一个 Metis(数据科学夏令营)中的编程挑战题目,同时给了我一个机会来制作我的第一个 Python 动画。看看结果的动图:
640?wx_fmt=gif

这篇文章的重点还是主要放在 python 中如何用 Matploylib 制作动画。
但如果你不太熟悉模拟游戏的话(它更像是可以看的模拟动画,而非可以玩的游戏),我来给大家介绍一下规则:
  • 一开始先设置一个 N×N 的网格(我的动画中用的是 50×50 );

  • 接着随机地向格子中填充“小细胞”(一开始随机地从 2500 个格子中选取 1500 个进行填充);

  • 如果邻居小细胞少于等于 1 个,那格子中的小细胞会死掉;

  • 如果邻居大于等于 4 个的也会死掉;

  • 只有 2 个或 3 个邻居时可以生存;

  • 空的格子中如果正好有 3 个邻居,则会长出 1 个新的“小细胞”;

—  2 

建立网格

我们首先导入所需的库。
import time	
from IPython import display	
import matplotlib.pyplot as plt	
import matplotlib.animation as animation
我们会利用  Matploylib 动画模块中的 FuncAnimation() 函数。 FuncAnimation()是通过多次调用一个函数并逐次更新图片来实现让图片动起来的。 我们来一步步地实现这个过程。
但首先,我们需要先初始化我们的网格。下面的几行代码用来存储我们输入的数据:
  • 我们需要一个 50×50 大小的网格;

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

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

# Input variables for the board	
boardsize = 50        # board will be X by X where X = boardsize	
pad = 2               # padded border, do not change this!	
initial_cells = 1500  # this number of initial cells will be placed	# in randomly generated positions
接下来我们随机地生成一系列“小细胞”的初始坐标(上面我们选择了 1500 个)。把这些坐标存储在  pos_list 变量中。
# Get a list of random coordinates so that we can initialize	
# board with randomly placed organisms	
pos_list = []	
for i in range(initial_cells):	pos_list.append([random.randint(1, boardsize),	random.randint(1, boardsize)])
然后我们是时候该初始化网格了。我们会用一组叫  my_board 的 numpy 序列来代表我们的网格——我们先生成一个 52×52 数值为 0 的矩阵序列作为开始(比 50×50 大是由于增加了空白边缘),然后调用  init_board() 函数来根据  pos_list 中的坐标把“小细胞”填充到网格中。辅助函数的具体细节我不再展开讲了,不过我把他们都整理到我的 Github 上了。
# Initialize the board	
my_board = np.zeros((boardsize+pad, boardsize+pad))	
my_board = init_board(pos_list, my_board)

—  3 

制作网格动画

这是我们最期待的部分——动画!首先,我们需要完善一些配置。下面的几行代码用来生成展示我们动画的  mtplotlib 图框。
# Required line for plotting the animation	
%matplotlib notebook	
# Initialize the plot of the board that will be used for animation	
fig = plt.gcf()
接下来制作我们的第一帧。  mtplotlib 中的  imshow() 函数可以接收一组  numpy 矩阵然后返回一张图片。很酷吧!
# Show first image - which is the initial board	
im = plt.imshow(my_board)	
plt.show()
传入  imshow() 的变量是我们的初始的网格  my_board。生成的图片长这样:
640?wx_fmt=png
现在我们需要写一个可以给  FuncAnimation() 调用的辅助函数。  animate() 函数接受一帧画面作为输入充当计数器。这个画面计数器就是  FuncAnimation() 和  animate() 函数沟通的桥梁——在每一个时间点(也就是每一帧),它都会调用一次  animate()。然后  animate() 会逐次使用辅助函数  update_board() 来对网格进行迭代。最后,  set_data() 函数将图片更新为迭代后的网格,这就完成了。
# Helper function that updates the board and returns a new image of	
# the updated board animate is the function that FuncAnimation calls	
def animate(frame):	im.set_data(update_board(my_board))	return im,
一切顺利!我们准备调用  FuncAnimation() 函数了。注意输入的参数:
  • fig 是我们在前面创建的用来装载我们的动画的图形变量;

  • animate 是 FuncAnimation() 用画面计数器进行沟通的函数(自动传入,不需要特别声明)

  • frames 表示我们希望动画持续多少帧,在这里我们想要动画的长度为 200 帧;

  • interval 表示每一帧之间间隔的毫秒数。我们想要每帧之间间隔 50 毫秒。
# This line creates the animation	
anim = animation.FuncAnimation(fig, animate, frames=200,	interval=50)
就这么简单!不是很难吧?为了庆祝我们成功制作动画,我再送大家一个动画:
640?wx_fmt=gif

—  4 

总结

希望这篇文章能帮到大家。在结束之前,让我来帮助大家脑补更多我们今天学到的动画功能在数据科学上的应用:
  • 一个个地画出蒙特卡洛模拟数据,你能观察到最终的分布是如何逐步形成的;

  • 按顺序遍历时间序列数据,可以描绘你的模型或数据在新的观察角度下有什么表现;

  • 当你改变输入参数时,比如族群数,可以展现你的算法是如何划分族群的;

  • 根据时间或不同的数据子集生成关联热力图,用于观察不同的样本是如何影响你的模型的预期参数的。


Github: https://github.com/yiuhyuk/game_of_life
来源:https://towardsdatascience.com/spice-up-your-python-visualizations-with-matplotlib-animations-d437d7e98e67

本文来自公众号读者翻译,欢迎各位童鞋向公号投稿,点击下面图片了解详情!

译者简介
Hanz,港科大硕士,三年互联网产品经理。自学数据分析及人工智能技术,现于美国梅奥医院医学信息实验室担任研究员。

-------------------End-------------------

640?wx_fmt=jpeg

「Python数据之道 」建立了读者交流群,大家可以添加管理员微信进行加群
扫描添加好友
回复 “ 资源
  • 可视化神器推荐(Plotly Express)

Python数据之道  
让数据更有价值
回复 “600”,
获取《Python知识手册》
同学们,支持就请右下角点640?wx_fmt=gif

这篇关于巧用 Matplotlib 动画,让你的 Python 可视化大放异彩的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!



http://www.chinasem.cn/article/625175

相关文章

使用Python实现矢量路径的压缩、解压与可视化

《使用Python实现矢量路径的压缩、解压与可视化》在图形设计和Web开发中,矢量路径数据的高效存储与传输至关重要,本文将通过一个Python示例,展示如何将复杂的矢量路径命令序列压缩为JSON格式,... 目录引言核心功能概述1. 路径命令解析2. 路径数据压缩3. 路径数据解压4. 可视化代码实现详解1

python获取网页表格的多种方法汇总

《python获取网页表格的多种方法汇总》我们在网页上看到很多的表格,如果要获取里面的数据或者转化成其他格式,就需要将表格获取下来并进行整理,在Python中,获取网页表格的方法有多种,下面就跟随小编... 目录1. 使用Pandas的read_html2. 使用BeautifulSoup和pandas3.

Python装饰器之类装饰器详解

《Python装饰器之类装饰器详解》本文将详细介绍Python中类装饰器的概念、使用方法以及应用场景,并通过一个综合详细的例子展示如何使用类装饰器,希望对大家有所帮助,如有错误或未考虑完全的地方,望不... 目录1. 引言2. 装饰器的基本概念2.1. 函数装饰器复习2.2 类装饰器的定义和使用3. 类装饰

Python 交互式可视化的利器Bokeh的使用

《Python交互式可视化的利器Bokeh的使用》Bokeh是一个专注于Web端交互式数据可视化的Python库,本文主要介绍了Python交互式可视化的利器Bokeh的使用,具有一定的参考价值,感... 目录1. Bokeh 简介1.1 为什么选择 Bokeh1.2 安装与环境配置2. Bokeh 基础2

如何使用 Python 读取 Excel 数据

《如何使用Python读取Excel数据》:本文主要介绍使用Python读取Excel数据的详细教程,通过pandas和openpyxl,你可以轻松读取Excel文件,并进行各种数据处理操... 目录使用 python 读取 Excel 数据的详细教程1. 安装必要的依赖2. 读取 Excel 文件3. 读

Python的time模块一些常用功能(各种与时间相关的函数)

《Python的time模块一些常用功能(各种与时间相关的函数)》Python的time模块提供了各种与时间相关的函数,包括获取当前时间、处理时间间隔、执行时间测量等,:本文主要介绍Python的... 目录1. 获取当前时间2. 时间格式化3. 延时执行4. 时间戳运算5. 计算代码执行时间6. 转换为指

利用Python调试串口的示例代码

《利用Python调试串口的示例代码》在嵌入式开发、物联网设备调试过程中,串口通信是最基础的调试手段本文将带你用Python+ttkbootstrap打造一款高颜值、多功能的串口调试助手,需要的可以了... 目录概述:为什么需要专业的串口调试工具项目架构设计1.1 技术栈选型1.2 关键类说明1.3 线程模

Python ZIP文件操作技巧详解

《PythonZIP文件操作技巧详解》在数据处理和系统开发中,ZIP文件操作是开发者必须掌握的核心技能,Python标准库提供的zipfile模块以简洁的API和跨平台特性,成为处理ZIP文件的首选... 目录一、ZIP文件操作基础三板斧1.1 创建压缩包1.2 解压操作1.3 文件遍历与信息获取二、进阶技

Python Transformers库(NLP处理库)案例代码讲解

《PythonTransformers库(NLP处理库)案例代码讲解》本文介绍transformers库的全面讲解,包含基础知识、高级用法、案例代码及学习路径,内容经过组织,适合不同阶段的学习者,对... 目录一、基础知识1. Transformers 库简介2. 安装与环境配置3. 快速上手示例二、核心模

Python正则表达式语法及re模块中的常用函数详解

《Python正则表达式语法及re模块中的常用函数详解》这篇文章主要给大家介绍了关于Python正则表达式语法及re模块中常用函数的相关资料,正则表达式是一种强大的字符串处理工具,可以用于匹配、切分、... 目录概念、作用和步骤语法re模块中的常用函数总结 概念、作用和步骤概念: 本身也是一个字符串,其中