Matplotlib Python中的Cmap

Question

我正在尝试使用matplotlib绘制Mandelbrot集。我用一种颜色完成了它，但现在我试图根据迭代次数输入一些cmap。所以，我的代码看起来像：

import random
import matplotlib.pyplot as plt


elem = 10000
x = []
y = []
colors = []

def iteration(x):
    n = 0
    result = 0
    while n < 1000: # max iterations
        result = pow(result, 2) + c
        if abs(result) > 2:
            return n           
        n += 1
    return n

for i in range(elem):
    c = complex(random.uniform(-2.5,1),random.uniform(-1.5,1.5))
    x.append(c.real)
    y.append(c.imag)
    colors.append(iteration(c))


plt.scatter(x, y, ',', c = colors, cmap= 'magma', vmin = 0, vmax = 1000)
plt.axis('equal')
plt.axis([-2,1,-1.5,1.5])
plt.title('Mandelbrot Set', y = 1.05)
plt.ylabel('Im')
plt.xlabel('Re')
plt.colorbar()
plt.show()

我不确定如何在plt.scatter中定义c，vmin和vmax。我想，例如，如果n是1000（最大迭代）黑点，另一种颜色，如果n = 0。

Answer 1

我确实认为您的代码运行良好。只需要定义标记。

plt.scatter(x, y, marker=',', c = colors, cmap= 'magma', vmin = 0, vmax = 1000)