manim 生成勾股树

下面是一个使用 manim 生成勾股树的例子：

from manim import *
import math

class PythagoreanTree(Scene):
    def construct(self):
        # 定义勾股树的参数
        angle = math.pi/4
        ratio = math.sin(angle)

        # 定义勾股树的函数
        def pythagorean_tree(start, angle, length, depth):
            if depth > 0:
                end = start + length * complex(math.cos(angle), math.sin(angle))
                line = Line(start, end)

                self.play(Create(line))

                left = pythagorean_tree(end, angle - math.pi/4, length * ratio, depth - 1)
                right = pythagorean_tree(end, angle + math.pi/4, length * ratio, depth - 1)

                return [line, left, right]
            else:
                return []

        # 生成勾股树
        tree = pythagorean_tree(complex(0, 0), angle, 4, 6)

        # 显示勾股树
        self.wait(1)

这个例子定义了一个 PythagoreanTree 类，该类继承自 manim 的 Scene 类。在 PythagoreanTree 类中，我们定义了一个 pythagorean_tree 函数，该函数使用递归的方式生成勾股树。然后，我们在 construct 方法中调用这个函数，并使用 manim 的 Create 方法显示勾股树。在这个例子中，我们生成了一个深度为 6 的勾股树。你可以根据需要调整 angle、ratio、length 和 depth 参数来生成不同的勾股树。