Python - 颜色根据坐标变化？

Question

我已经成功地编写了一个绘制mandelbrot集的代码，但是现在我尝试根据迭代之前的不同颜色设置不同的颜色，但是我无法让它工作。这是我的代码，我不知道它有什么问题......

from math import *
import matplotlib.pyplot as plt


z=complex(0,0)

c=complex(2,2)
X=[]
Y=[]
Z=[]
p=1000
ite=30
a=-(4/p+4j/p)
for m in range(p):
    c=complex(2,c.imag+a.imag)
    for i in range(p):
        c=c+a.real
        z=complex(0,0)
        b=1
        for i in range(ite):
            b=b-0.005
            z=z**2+c
            if abs(z)>2:
                break

        X=X+[c.real]
        Y=Y+[c.imag]
        Z=Z+[b/30]


    print(round((m/p)*100),"%")
colors = Z
plt.scatter(X,Y,'.',c=[colors])
plt.show()

提前感谢！