使用Python多进程生成Mandelbrot集图像时程序冻结问题

核心原因 #

你的程序冻结本质是主进程在读取队列时陷入无限阻塞，和分辨率阈值相关是因为：当分辨率≤87px时，单条带数据量小，队列读取逻辑的隐性问题未触发；≥88px时，数据量增大后，主进程的读取循环失去终止条件，一直等待不存在的队列数据。

常见触发场景：

1. 主进程用while True循环读取队列，但仅靠len(strips)判断终止，若某次队列读取顺序异常或计数错误，会导致循环无法退出。
2. 使用queue.join()但未在子进程中调用queue.task_done()，主进程会永久等待队列任务完成的信号。
3. 依赖queue.empty()判断是否停止读取，但该方法不可靠（检查时队列非空，读取时可能已空）。

解决方法 #

方案1：明确读取次数（最可靠）

因为每个子进程对应生成1条带，主进程只需读取进程数次队列数据即可，无需依赖队列状态或计数判断。

修正后的核心代码示例：

import multiprocessing
from PIL import Image

def compute_mandelbrot(x, y, max_iter):
    c = complex(x, y)
    z = 0
    for i in range(max_iter):
        z = z*z + c
        if abs(z) > 2:
            return i
    return max_iter

def generate_strip(y_start, y_end, width, height, x_range, y_range, max_iter, queue):
    strip = []
    x_min, x_max = x_range
    y_min, y_max = y_range
    for y in range(y_start, y_end):
        row = []
        y_norm = y_min + (y / height) * (y_max - y_min)
        for x in range(width):
            x_norm = x_min + (x / width) * (x_max - x_min)
            row.append(compute_mandelbrot(x_norm, y_norm, max_iter))
        strip.append(row)
    queue.put((y_start, strip))

def main():
    width, height = 88, 88
    x_range = (-2.0, 1.0)
    y_range = (-1.5, 1.5)
    max_iter = 255
    num_processes = multiprocessing.cpu_count()
    strip_height = height // num_processes

    queue = multiprocessing.Queue()
    processes = []

    # 启动子进程
    for i in range(num_processes):
        y_start = i * strip_height
        y_end = (i+1)*strip_height if i != num_processes-1 else height
        p = multiprocessing.Process(
            target=generate_strip,
            args=(y_start, y_end, width, height, x_range, y_range, max_iter, queue)
        )
        processes.append(p)
        p.start()

    # 等待所有子进程完成计算并写入队列
    for p in processes:
        p.join()

    # 固定读取次数，避免无限阻塞
    strips = {}
    for _ in range(num_processes):
        y_start, strip = queue.get()
        strips[y_start] = strip

    # 合并条带生成图像
    full_image = []
    for y_start in sorted(strips.keys()):
        full_image.extend(strips[y_start])
    
    img = Image.new('L', (width, height))
    img.putdata([pixel for row in full_image for pixel in row])
    img.save('mandelbrot.png')

if __name__ == '__main__':
    main()

方案2：使用结束标记

若必须用循环读取，可让所有子进程完成后，向队列放入一个特殊结束标记（如None），主进程读取到标记后终止循环：

• 在所有子进程join()后，执行queue.put(None)
• 主进程读取逻辑改为：

  strips = {}
  while True:
      data = queue.get()
      if data is None:
          break
      y_start, strip = data
      strips[y_start] = strip
  

方案3：修复queue.join()使用

若你之前用了queue.join()，必须在子进程的queue.put()后调用queue.task_done()，否则主进程会永久阻塞：

def generate_strip(..., queue):
    # 生成条带逻辑
    queue.put((y_start, strip))
    queue.task_done()  # 必须添加该行

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