Interferenza di onde circolari con la calcolatrice grafica

from math import *
from kandinsky import *

for y in range(222):
  for x in range(161):
    z1 = cos(2*pi/20*sqrt((x+40)**2+y**2))
    z2 = cos(2*pi/20*sqrt((x-40)**2+y**2))
    z = z1+z2
    c = color(0,0,128+int(z*63))
    set_pixel(160-x,y,c)
    set_pixel(160+x,y,c)

import numpy as np
import matplotlib.pyplot as plt
from math import pi

x = np.arange(-5, 5, 0.05)
y = np.arange(-4, 4, 0.05)
x, y = np.meshgrid(x, y)
x1 = np.sqrt((x-1)**2 + y**2)
x2 = np.sqrt((x+1)**2 + y**2)
z = np.cos(2*pi*x1)+np.cos(2*pi*x2)

fig, ax = plt.subplots()
fig.canvas.set_window_title('Interferenza di onde circolari')
ax.title.set_text('Interferenza di onde circolari')
ax.set(xlabel='X', ylabel='Y')
ax.contour(x, y, z)

plt.show()