python绘制Mandelbrot分形 详细注释版

import numpy as np
import matplotlib.pyplot as plt 
def mandelbrot( h,w, maxit=5 ):
    """Returns an image of the Mandelbrot fractal of size (h,w)."""
    # 通过复数的方式, 生成x,y坐标
    y,x = np.ogrid[ -1.4:1.4:h*1j, -2:0.8:w*1j ]                      
    c = x+y*1j
    z = c 
    # 对divtime设置初始值为最大值20
    divtime = maxit + np.zeros(z.shape, dtype=int)              
    for i in range(maxit):
        # Mandelbrot特点: 如果 c in M，则 |c| <= 2; 反过来不一定成立
        z = z**2 + c                              
        # 找出大于2的点                          
        diverge = z*np.conj(z) > 2**2      
        # divtime==maxit表示还没有修改过divtime的点, 然后再与大于2的点求交集, 找到位置
        div_now = diverge & (divtime==maxit)               
        # 跟据位置在divtime中设置迭代次数, 用以生成不同的颜色  
        divtime[div_now] = i   
        # 刷一遍所有已经大于2的值的点, 这里可以是任何值, 只要z**2 + c不溢出就行
        z[diverge] =   2                                                    
    return divtime
plt.imshow(mandelbrot(4,4))
plt.show()

原理参考:
https://blog.csdn.net/q1007729991/article/details/79510494