python manim实现排序算法动画示例


Posted in Python onAugust 14, 2022

什么是 manim

Manim 是一个用于精确编程动画的引擎,专为创建解释性数学视频而设计。

注意,有两个主要版本的 manim。该存储库最初是 3Blue1Brown 的作者的个人项目,目的是为这些视频制作动画,此处提供了视频专用代码。2020 年,一群开发人员将其分叉成现在的社区版,目标是更稳定、更好地测试、更快地响应社区贡献,以及更友好地开始使用。

主要版本如下:

冒泡排序介绍

本文就使用 manim 来实现一个冒泡排序的动画,首先来了解下什么是冒泡排序

冒泡排序(Bubble Sort)也是一种简单直观的排序算法。它重复地走访过要排序的数列,一次比较两个元素,如果他们的顺序错误就把他们交换过来。走访数列的工作是重复地进行直到没有再需要交换,也就是说该数列已经排序完成。这个算法的名字由来是因为越小的元素会经由交换慢慢"浮"到数列的顶端。

算法步骤

  • 比较相邻的元素。如果第一个比第二个大,就交换他们两个。
  • 对每一对相邻元素作同样的工作,从开始第一对到结尾的最后一对。这步做完后,最后的元素会是最大的数。
  • 针对所有的元素重复以上的步骤,除了最后一个。
  • 持续每次对越来越少的元素重复上面的步骤,直到没有任何一对数字需要比较。

初始化元素

比如我们需要排序数组为: [4,2,3,1,5]

首先,需要在 manim 场景上初始化我们的需要排序的所有元素,这里用矩形来表示。

在 manim 中,可以用 Rectangle 来初始化矩形,然后我们通过设置元素不同的高度来表示不同的元素大小。

  • main.py
from manimlib import *
class Test(Scene):
    def construct(self):
        COLOR = [BLUE, GREEN, RED, PINK, ORANGE, MAROON_B, TEAL, PURPLE_B, GREY_BROWN]
        arr = [4,2,3,1,5]
        g = VGroup()
        for i in range(len(arr)):
            r1=Rectangle(width=1,height=arr[i],fill_color=COLOR[i%len(COLOR)],fill_opacity=1)
            t1=Text(str(arr[i])).scale(0.5)
            rec = VGroup(r1,t1)
            g.add(rec)
        g.arrange(RIGHT,aligned_edge=DOWN)
        self.add(g)
        self.wait()

使用下面的命令运行上面的代码:

manimgl main.py BubbleSort

ManimGL v1.6.1
[11:27:18] INFO     Using the default configuration file, which you can modify in `/Users/zheng/anaconda3/envs/manim/lib/python3.10/site-packages/manimlib/default_config.yml`                                                                                         config.py:265
           INFO     If you want to create a local configuration file, you can create a file named `custom_config.yml`, or run `manimgl --config`                                                                                                                       config.py:266
[11:27:20] INFO     Tips: You are now in the interactive mode. Now you can use the keyboard and the mouse to interact with the scene. Just press `q` if you want to quit.

运行后,就会出现一个窗口显示如下画面。

python manim实现排序算法动画示例

代码说明

上面代码中,通过继承父类 Scene 然后重新父类的 construct 来构建一个场景。

然后在场景中添加了矩形(Rectangle)和文本(Text),并且将这两个元素添加到了 VGroup 类中。

再用一个 VGroup 来包含所有的 VGroup,通过调用 arrange 方法来排列这些元素。第一个 RIGHT 参数表示所有元素向右依次排列,aligned_edge 表示对齐的边,这里我们传入 DOWN 将底边对齐。

最后使用 self.add() 方法把 VGroup 添加到场景中。

Rectangle 类定义了矩形的创建,更多图形可以查看 docs.manim.org.cn/documentati…

元素交换动画

通过算法步骤的第一步:比较相邻的元素。如果第一个比第二个大,就交换他们两个。就涉及到了交换的动画。

一开始,我用 manim 提供的 CyclicReplace 方法来交换两个元素。效果如下:

self.play(CyclicReplace(g[0], g[1]))
self.wait()

python manim实现排序算法动画示例

交换是交换了,但是交换后对齐的边变成了顶部对齐了,不符合预期。于是继续查看文档,最终决定使用元素的 target 属性来进行交换动画的制作。

上面我们要交换 g(0)g(1) 两个元素,所以我们定义这两个交换元素的 target

g[0].generate_target()
g[0].target.next_to(g[1],ORIGIN,aligned_edge=DOWN)
g[1].generate_target()
g[1].target.next_to(g[0],ORIGIN,aligned_edge=DOWN)

generate_target() 表示生成元素的 target, next_to() 表示将元素移动到指定的位置。

比如 g[0],我们先生成元素的 target,然后操作 target 将元素通过 next_to 方法移动到 g[1] 的位置。其中 ORIGIN 表示 g[1] 的所在位置。 我们对 g[1] 的元素也做类似的操作。

然后使用 MoveToTarget 来将元素转换到定义的 target 上,通过调用 self.play() 方法来播放动画。

self.play(MoveToTarget(g[0]),MoveToTarget(g[1]))

python manim实现排序算法动画示例

嗯~完美符合预期。

实现代码

根据上面的知识点,接下来就可以编写一个冒泡排序的动画了。

这里在初始化场景元素时,额外添加了一个数组来存放所有场景元素,因为在交换元素位置后,也要交换对应索引下的元素,如果直接用 VGroup 来交换时,会出现问题。

self.g[j],self.g[j+1] = self.g[j+1],self.g[j]
TypeError: 'VGroup' object does not support item assignment

所以用额外的数组去接收。

还添加了一个 Indicate 方法,当涉及到对应交换的元素时,会做一个类似对焦的动作。

from manimlib import *
class BubbleSort(Scene):
    def construct(self):
        self.COLOR = [BLUE, GREEN, RED, PINK, ORANGE, MAROON_B, TEAL, PURPLE_B, GREY_BROWN]
        self.bubbleSort([4,2,3,1,5])
    def init_vmobj(self,arr):
        '''
            初始化场景元素
        '''
        self.vmArr = []
        g = VGroup()
        for i in range(len(arr)):
            r1=Rectangle(width=1,height=arr[i]/2,fill_color=self.COLOR[i%len(self.COLOR)],fill_opacity=1)
            t1=Text(str(arr[i])).scale(0.5)
            rec = VGroup(r1,t1)
            self.vmArr.append(rec)
            g.add(rec)
        g.arrange(RIGHT,aligned_edge=DOWN)
        self.add(g)
        self.wait()
    def bubbleSort(self,arr):
        '''
            冒泡排序
        '''
        self.init_vmobj(arr)
        for i in range(1, len(arr)):
            for j in range(0, len(arr)-i):
                self.play(Indicate(self.vmArr[j]))
                self.play(Indicate(self.vmArr[j+1],color=RED))
                if arr[j] > arr[j+1]:
                    arr[j], arr[j + 1] = arr[j + 1], arr[j]
                    self.cyc_move(self.vmArr[j],self.vmArr[j+1])
                    self.vmArr[j],self.vmArr[j+1] = self.vmArr[j+1],self.vmArr[j]
        return arr
    def cyc_move(self,vm1,vm2):
        '''
            交换两个元素位置
        '''
        vm1.generate_target()
        vm1.target.next_to(vm2,ORIGIN,aligned_edge=DOWN)
        vm2.generate_target()
        vm2.target.next_to(vm1,ORIGIN,aligned_edge=DOWN)
        self.play(MoveToTarget(vm1),MoveToTarget(vm2))
        self.wait()

python manim实现排序算法动画示例

以上就是python manim实现排序算法动画示例的详细内容

本文不介绍 manim 的安装教程,需要安装教程的请参考:docs.manim.org.cn/getting_sta

更多关于python manim排序算法动画的资料请关注三水点靠木其它相关文章!

Python 相关文章推荐
Django应用程序中如何发送电子邮件详解
Feb 04 Python
Python使用正则表达式过滤或替换HTML标签的方法详解
Sep 25 Python
python之matplotlib学习绘制动态更新图实例代码
Jan 23 Python
30秒轻松实现TensorFlow物体检测
Mar 14 Python
Python中偏函数用法示例
Jun 07 Python
为什么str(float)在Python 3中比Python 2返回更多的数字
Oct 16 Python
Pycharm更换python解释器的方法
Oct 29 Python
对Python3中列表乘以某一个数的示例详解
Jul 20 Python
使用PyCharm进行远程开发和调试的实现
Nov 04 Python
Python中url标签使用知识点总结
Jan 16 Python
python操作yaml说明
Apr 08 Python
用python制作个音乐下载器
Jan 30 Python
Python 操作pdf pdfplumber读取PDF写入Exce
Aug 14 #Python
Python使用plt.boxplot()函数绘制箱图、常用方法以及含义详解
Aug 14 #Python
基于Python实现nc批量转tif格式
Aug 14 #Python
LyScript实现绕过反调试保护的示例详解
Aug 14 #Python
LeetCode189轮转数组python示例
Aug 05 #Python
python语言中pandas字符串分割str.split()函数
Aug 05 #Python
python绘制云雨图raincloud plot
Aug 05 #Python
You might like
解析php中static,const与define的使用区别
2013/06/18 PHP
ThinkPHP 3.2 数据分页代码分享
2014/10/14 PHP
php实现修改新闻时删除图片的方法
2015/05/12 PHP
php实现登陆模块功能示例
2016/10/20 PHP
对xmlHttp对象方法和属性的理解
2011/01/17 Javascript
jquery 提交值不为空的元素示例代码
2013/05/10 Javascript
浅析document.createDocumentFragment()与js效率
2013/07/08 Javascript
多次注册事件会导致一个事件被触发多次的解决方法
2013/08/12 Javascript
javascript动态创建表格及添加数据实例详解
2015/05/13 Javascript
Javascript实现鼠标右键特色菜单
2015/08/04 Javascript
JavaScript简单判断复选框是否选中及取出值的方法
2015/08/13 Javascript
再谈Javascript中的异步以及如何异步
2016/08/19 Javascript
JavaScript实现输入框与清空按钮联动效果
2016/09/09 Javascript
微信小程序 window_x64环境搭建
2016/09/30 Javascript
jquery中封装函数传递当前元素的方法示例
2017/05/05 jQuery
微信小程序使用modal组件弹出对话框功能示例
2017/11/29 Javascript
JS实现DOM删除节点操作示例
2018/04/04 Javascript
微信小程序scroll-view仿拼多多横向滑动滚动条
2020/04/21 Javascript
Node.js npm命令运行node.js脚本的方法
2018/10/10 Javascript
Intellij IDEA搭建vue-cli项目的方法步骤
2018/10/20 Javascript
Vue 实时监听窗口变化 windowresize的两种方法
2018/11/06 Javascript
微信小程序通过一个json实现分享朋友圈图片
2019/09/03 Javascript
vue打包npm run build时候界面报错的解决
2020/08/13 Javascript
axios解决高并发的方法:axios.all()与axios.spread()的操作
2020/11/09 Javascript
python3.7 sys模块的具体使用
2019/07/22 Python
Django框架下静态模板的继承操作示例
2019/11/08 Python
python爬虫开发之使用Python爬虫库requests多线程抓取猫眼电影TOP100实例
2020/03/10 Python
Python urlencode和unquote函数使用实例解析
2020/03/31 Python
编码实现字符串转整型的函数
2012/06/02 面试题
无工作经验者个人求职信范文
2013/12/22 职场文书
项目申请汇报材料
2014/08/16 职场文书
2014年人力资源工作总结
2014/11/19 职场文书
《卖火柴的小女孩》教学反思
2016/02/19 职场文书
Java8中接口的新特性使用指南
2021/11/01 Java/Android
nginx location 带斜杠【 / 】与不带的区别
2022/04/13 Servers
浅谈为什么我的 z-index 又不生效了
2022/07/15 HTML / CSS