I am trying to plot Mandelbrot set using matplotlib. I've done it with one color, but now i am trying to put in some cmap depending on number of iterations. So, my code looks like:
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()
I am not sure how to define c, vmin and vmax in plt.scatter. I would like to have, for example if n is 1000 (max iteration) black dot and another color if n = 0.
The technical post webpages of this site follow the CC BY-SA 4.0 protocol. If you need to reprint, please indicate the site URL or the original address.Any question please contact:yoyou2525@163.com.