def f(x, y):
    return pylab.cos(x) + pylab.sin(y)
xx = pylab.linspace(-5, 5, 100)
yy = pylab.linspace(-5, 5, 100)
zz = pylab.zeros([len(xx), len(yy)])
for i in range(len(xx)):
    for j in range(len(yy)):
        zz[j, i] = f(xx[i], yy[j])
pylab.pcolor(xx, yy, zz)
pylab.show()
Как построить 3D-функцию в виде 2D-цветовой карты в python?