Python-颜色根据坐标变化？

我已经成功地编写了一个代码来绘制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()

---

提前谢谢

它说颜色数组必须是两维的当提出问题时，最好尽量简化。所以这里的问题不是关于Mandelbrot集，而是改变第三个变量的颜色。