MatPlotLib

http://matplotlib.sourceforge.net/examples/pylab_examples/pcolor_demo.html

 

pylab_examples example code: pcolor_demo.py¶

[source code, hires.png, pdf]

#!/usr/bin/env python

"""

See pcolor_demo2 for an alternative way of generating pcolor plots

using imshow that is likely faster for large grids

"""

from __future__ import division

from matplotlib.patches import Patch

from pylab import *



def func3(x,y):

    return (1- x/2 + x**5 + y**3)*exp(-x**2-y**2)





# make these smaller to increase the resolution

dx, dy = 0.05, 0.05



x = arange(-3.0, 3.0001, dx)

y = arange(-3.0, 3.0001, dy)

X,Y = meshgrid(x, y)



Z = func3(X, Y)

pcolor(X, Y, Z)

colorbar()

axis([-3,3,-3,3])



show()

Keywords: python, matplotlib, pylab, example, codex (see Search examples)