计算没有复杂对象的 Mandelbrot 集

Calculate Mandelbrot set without complex objects

我正在尝试计算 Python 3.6 中的 Mandelbrot 集,但我不想使用复杂的对象来计算它。有人有 getIterationCount(x, y) 功能吗?

我尝试将 java 代码重写为 python,但没有成功。

def getIterationCount(x, y):
    maxiter = 100
    z = complexe(x, y)
    c = z
    for n in range(0, maxiter):
        if abs(z) > 2:
            return n
        z = z*z + c
    return maxiter

如果你只想使用实数,我可以帮你写:

def getIterationCount(ca,cb):
    maxiter = 100
    za, zb = ca,cb
    for n in range(0, maxiter):
        if za**2+zb**2 > 4:
            return n
        za, zb = za*za - zb*zb + ca, 2*za*zb + cb

    return maxiter