Numpy et fractale de Mandelbrot

[Pierre Levy]

Bonjour à toutes et à tous et bonne année

J'ai trouvé dans le livre "Numpy cookbook" le code suivant pour dessiner l'ensemble de Mandelbrot.

Code :

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
# -*- coding: utf-8 -*-
import numpy
import matplotlib.pyplot
 
import sys
 
if(len(sys.argv) != 2):
  print "Please input the number of  iterations for the fractal"
  sys.exit()
ITERATIONS = int(sys.argv[1])
SIZE=1024
MAX_COLOR = 255.
x_min, x_max = -2.5, 1
y_min, y_max = -1.75, 1.75
# Initialize arrays
x, y = numpy.meshgrid(numpy.linspace 
  (x_min, x_max, SIZE),  
  numpy.linspace(y_min, y_max, SIZE))
c = x + 1j * y
z = c.copy()
fractal = numpy.zeros(z.shape,  
  dtype=numpy.uint8) + MAX_COLOR
# Generate fractal
for n in range(ITERATIONS):
#    print n
    mask = numpy.abs(z) <= 4 
    z[mask] = z[mask] ** 2 +  c[mask]
    fractal[(fractal == MAX_COLOR) &  
      (-mask)] = (MAX_COLOR - 1) * n / ITERATIONS
# Display the fractal
matplotlib.pyplot.imshow(fractal)
matplotlib.pyplot.title('Mandelbrot')
matplotlib.pyplot.axis('on')
matplotlib.pyplot.colorbar()
matplotlib.pyplot.axis('off')
matplotlib.pyplot.title('Mandelbrot')
matplotlib.pyplot.show()

Ce code fonctionne parfaitement et me semble très efficace mais j'ai un gros problème de compréhension de la boucle for et en particulier des lignes 28 et 29.

Si une âme charitable et experte pouvait m'expliquer je l'en remercie par avance.