在Mandelbrot 集中“缩放”特定区域

import complex

def mandelbrot_zoom(uleft, lright, size, n):
    """
    Generate a Mandelbrot set image with a zoomed-in area.

    Args:
    uleft: Complex number representing the upper-left corner of the zoomed-in area.
    lright: Complex number representing the lower-right corner of the zoomed-in area.
    size: Tuple representing the size of the image (width, height).
    n: Maximum number of iterations to check for divergence.

    Returns:
    A list of lists of colors, where each color represents a pixel in the image.
    """

    # Calculate the width and height of the zoomed-in area.
    xwidth = lright.real - uleft.real
    ywidth = uleft.imag - lright.imag

    # Create a list to store the colors of the pixels.
    image = [[0 for _ in range(size[0])] for _ in range(size[1])]

    # Iterate over each pixel in the image.
    for y in range(size[1]):
        for x in range(size[0]):
            # Convert the pixel coordinates to complex coordinates.
            coords = (uleft.real + (x / size[0]) * xwidth,
                     uleft.imag - (y / size[1]) * ywidth)

            # Initialize the complex number z and the number of iterations.
            z = complex(0, 0)
            trials = 0

            # Iterate until z diverges or the maximum number of iterations is reached.
            while abs(z) <= 2.0 and trials < n:
                z = z**2 + coords
                trials += 1

            # Set the color of the pixel based on the number of iterations.
            image[y][x] = trials

    return image

# Example usage:
uleft = complex(-0.75, 0.1)
lright = complex(-0.5, -0.1)
size = (500, 500)
n = 64

image = mandelbrot_zoom(uleft, lright, size, n)