Matplotlib 3d colour plot - is it possible?

I've googled around for this and am unable to find anything which suggests that it may not exist, but I wanted to ask stackoverflow.

I'm using python, together with m开发者_运维技巧atplotlib and I'm looking to plot something like this: http://en.wikipedia.org/wiki/File:Spherical_harmonics.png

Essentially what I would call a 3d heatmap, a 3d surface with colours representing the magnitude of a function at that point. Similar to the link above I want to plot a colourised sphere, showing the value of the magnitude of this function at this point.

Is this possible

You can use the mplot3d toolkit, http://matplotlib.sourceforge.net/mpl_toolkits/mplot3d/index.html