PYTHON:

Use the file provided. It looks like this:

import numpy as np

import matplotlib.pyplot as plt

xmin, xmax = -2., 1.

ymin, ymax = -1.5, 1.5

num_pixels = 1000

# initial setup of calculation constant C for each pixel

X = np.linspace(xmin, xmax, num_pixels)[None,:]

Y = np.linspace(ymin, ymax, num_pixels)[:,None]

C = X + 1j * Y

# start value of Z is always 0

Z = np.zeros_like(C)

# P counts iterations, this is what we plot at the end

P = np.zeros_like(C, dtype=’uint8′) # unsigned int 0..255

# iteration of Z <- Z*Z + C

for i in range(120):

# print(f”Iteration {i}”)

# which elements are still “live”?

live = np.abs(Z) < 2.

# update live pixels with current iteration number

P[live] = i

# iterate

Z[live] = Z[live]*Z[live] + C[live]

plt.imshow(

P,

origin=’lower’,

extent=(X.min(), X.max(), Y.min(), Y.max())

)

plt.savefig(“mandelbrot.png”)

1)
Instead of everything happening in the main program, enter everything that is relevant in a function
almond (x, y, size, pixels, filename) which takes in 5 arguments and should use plt.savefig to save an image file. The function should not return anything.
The arguments x and y are coordinates of the lower left corner,
size is the distance from x / ymin to x / ymax,
pixels is the number of pixels in each direction, and
filename is the name of the file to be saved.