from kandinsky import set_pixel from math import * c=pi/2000 coul=[(0,0,0),(248,0,0),(248,252,0),(248,252,248)] def f(x,y): return 2000*exp(-x*x*4)*cos(3*y) for x in range(320): for y in range(222): xr,yr = (x-160)*c, (y-111)*c v=int(f(xr,yr))%4 set_pixel(x,y,coul[v])