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 相关文章推荐
python实现去除下载电影和电视剧文件名中的多余字符的方法
Sep 23 Python
python 排序算法总结及实例详解
Sep 28 Python
Python 中迭代器与生成器实例详解
Mar 29 Python
Python set常用操作函数集锦
Nov 15 Python
教你用 Python 实现微信跳一跳(Mac+iOS版)
Jan 04 Python
python2.x实现人民币转大写人民币
Jun 20 Python
使用python Telnet远程登录执行程序的方法
Jan 26 Python
python3.6根据m3u8下载mp4视频
Jun 17 Python
django框架使用方法详解
Jul 18 Python
Python学习笔记之列表和成员运算符及列表相关方法详解
Aug 22 Python
python  logging日志打印过程解析
Oct 22 Python
python定义类的简单用法
Jul 24 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 preg_filter执行一个正则表达式搜索和替换
2012/02/27 PHP
Java中final关键字详解
2015/08/10 PHP
laravel 查询数据库获取结果实现判断是否为空
2019/10/24 PHP
laravel实现图片上传预览,及编辑时可更换图片,并实时变化的例子
2019/11/14 PHP
学习YUI.Ext第七日-View&JSONView Part Two-一个画室网站的案例
2007/03/10 Javascript
javascript 获取元素位置的快速方法 getBoundingClientRect()
2009/11/26 Javascript
鼠标右击事件代码(asp.net后台)
2011/01/27 Javascript
JavaScript获取页面上某个元素的代码
2011/03/13 Javascript
实现局部遮罩与关闭原理及代码
2013/02/04 Javascript
老生常谈JavaScript 正则表达式语法
2016/08/20 Javascript
jQuery给指定的table动态添加删除行的操作方法
2016/10/12 Javascript
Bootstrap多级菜单的实现代码
2017/05/23 Javascript
angularjs中$http异步上传Excel文件方法
2018/02/23 Javascript
Angular6 Filter实现页面搜索的示例代码
2018/12/02 Javascript
angularjs http与后台交互的实现示例
2018/12/21 Javascript
基于Vue插入视频的2种方法小结
2019/04/02 Javascript
jQuery中DOM常见操作实例小结
2019/08/01 jQuery
js事件机制----捕获与冒泡机制实例分析
2020/05/22 Javascript
浅谈vue 多个变量同时赋相同值互相影响
2020/08/05 Javascript
Vue项目开发常见问题和解决方案总结
2020/09/11 Javascript
python retrying模块的使用方法详解
2019/09/25 Python
python pycharm最新版本激活码(永久有效)附python安装教程
2020/09/18 Python
python使用梯度下降和牛顿法寻找Rosenbrock函数最小值实例
2020/04/02 Python
python 用opencv实现霍夫线变换
2020/11/27 Python
五个2015 年最佳HTML5 框架
2015/11/11 HTML / CSS
Beach Bunny Swimwear官网:设计师泳装和性感比基尼
2019/03/13 全球购物
婚前保证书
2014/04/29 职场文书
植树节口号
2014/06/21 职场文书
小学生国庆65周年演讲稿范文(2篇)
2014/09/21 职场文书
2014学校领导四风问题对照检查材料思想汇报
2014/09/22 职场文书
小学五一劳动节活动总结
2015/02/09 职场文书
2015年员工工作表现评语
2015/03/25 职场文书
培训后的感想
2015/08/07 职场文书
2019已经过半,你知道年中工作总结该怎么写吗?
2019/07/03 职场文书
ICOM R71E和R72E图文对比解说
2022/04/07 无线电
Windows server 2003卸载和安装IIS的图文教程
2022/07/15 Servers