如何优化基于turtle的Mandelbrot集绘制代码以提升速度？

你的代码速度慢的核心原因是turtle库的逐点绘制（pen.dot()）效率极低，加上复数运算的额外开销、重复计算等问题，导致高分辨率下耗时极长。以下是针对性的优化方案，从“大幅提速”到“turtle内优化”分两种场景给出：

方案一：用Pillow直接生成图像（推荐，速度提升100+倍） #

turtle是为交互式绘图设计的，不适合处理数千上万像素的批量绘制。改用Pillow库直接生成像素图像，速度会快几个数量级，而且代码复杂度不高。

优化后代码 #

from PIL import Image

# 参数设置
WIDTH = 800
HEIGHT = 600
MAX_ITER = 80
# 曼德博集合的坐标范围
START_R, START_I = -2, -1
END_R, END_I = 1, 1

def mandelbrot_iter(real, imag, max_iter):
    # 用实部虚部分开计算，避免复数运算的额外开销
    x, y = 0.0, 0.0
    n = 0
    # 用x²+y² <=4代替abs(z)<=2，避免开平方运算
    while x*x + y*y <= 4 and n < max_iter:
        # 展开z² + c的实部和虚部计算
        x_new = x*x - y*y + real
        y_new = 2 * x * y + imag
        x, y = x_new, y_new
        n += 1
    return n

def generate_mandelbrot():
    # 创建黑色背景图像
    img = Image.new("RGB", (WIDTH, HEIGHT), (0, 0, 0))
    pixels = img.load()
    
    for x in range(WIDTH):
        for y in range(HEIGHT):
            # 计算当前像素对应的复数c
            real = START_R + (x / WIDTH) * (END_R - START_R)
            imag = START_I + (y / HEIGHT) * (END_I - START_I)
            # 迭代次数
            iter_count = mandelbrot_iter(real, imag, MAX_ITER)
            # 生成灰度颜色（也可以改成彩色）
            color = 255 - int(iter_count * 255 / MAX_ITER)
            pixels[x, y] = (color, color, color)
    
    # 显示图像
    img.show()
    # 可选：保存图像
    # img.save("mandelbrot.png")

generate_mandelbrot()

关键优化点 #

• 替换绘图库：Pillow直接操作像素数组，避免了turtle的GUI交互开销
• 迭代计算优化：
  • 用实部虚部分开计算，去掉复数类型的包装/解包开销
  • 用x*x + y*y <=4代替abs(z)<=2，避免开平方的耗时运算
• 无冗余计算：每个像素的坐标只计算一次，直接映射到图像像素

方案二：在turtle框架内优化（小幅提速，适合保留turtle场景） #

如果一定要用turtle，以下修改能让速度有所提升（但仍远不如Pillow）：

优化后代码 #

import turtle

screen = turtle.Screen()
screen.setup(width=1000, height=600, startx=0, starty=0)
screen.bgcolor('black')
screen.tracer(0, 0)  # 关闭自动刷新，手动控制
pen = turtle.Turtle()
pen.hideturtle()  # 隐藏画笔，减少绘制开销
pen.penup()

WIDTH = 800
HEIGHT = 600
MAX_ITER = 80

def mandelbrot_iter(real, imag, max_iter):
    x, y = 0.0, 0.0
    n = 0
    while x*x + y*y <= 4 and n < max_iter:
        x_new = x*x - y*y + real
        y_new = 2 * x * y + imag
        x, y = x_new, y_new
        n += 1
    return n

def generate_mandelbrot_set(StartR, StartI, EndR, EndI):
    screen.setworldcoordinates(StartR, StartI, EndR, EndI)
    # 预计算坐标增量，避免循环内重复计算
    delta_r = (EndR - StartR) / WIDTH
    delta_i = (EndI - StartI) / HEIGHT
    
    for x in range(WIDTH):
        real = StartR + x * delta_r
        # 先移动到当前列的顶部，再逐行绘制，减少goto的次数
        pen.goto(real, StartI)
        pen.pendown()
        for y in range(HEIGHT):
            imag = StartI + y * delta_i
            iter_count = mandelbrot_iter(real, imag, MAX_ITER)
            color = (255 - int(iter_count * 255 / MAX_ITER)) / 255
            pen.color(color, color, color)
            # 用forward(1)代替dot()，速度更快
            pen.forward(delta_i)
        pen.penup()
    screen.update()

generate_mandelbrot_set(-2,-1,1,1)
screen.mainloop()

关键优化点 #

• 迭代计算和方案一一致：去掉复数运算，避免开平方
• 减少turtle操作次数：
  • 隐藏画笔（hideturtle()），减少GUI渲染开销
  • 关闭自动刷新，最后一次性update()
  • 按列绘制，先移动到列顶部，再用forward()逐行画点，代替多次goto()
  • 预计算坐标增量（delta_r/delta_i），避免循环内重复除法运算
• 替换pen.dot()为pen.forward()：dot()的底层渲染开销远大于简单的线段绘制

内容的提问来源于stack exchange，提问作者Kris