python 3d图如何改变视角_如何在Python中将3D功能绘制为2D色彩图？

Are there any python libraries that will let me plot z = f(x,y) where z is represented as the color in a densely rasterized image (as opposed to the color of a bunch of scatterplot points) ? If so, what function do I use?

It looks like some of the contour functions in matplotlib.pyplot come close to what I want, but they draw contour lines and I don't want that.

解决方案

here's a concrete simple example (works also for functions which can't take matrix arguments for x and y):

# the function to be plotted

def func(x,y):

# gives vertical color bars if x is horizontal axis

return x

import pylab

# define the grid over which the function should be plotted (xx and yy are matrices)

xx, yy = pylab.meshgrid(

pylab.linspace(-3,3, 101),

pylab.linspace(-3,3, 111))

# indexing of xx and yy (with the default value for the

# 'indexing' parameter of meshgrid(..) ) is as follows:

#

# first index (row index) is y coordinate index

# second index (column index) is x coordinate index

#

# as required by pcolor(..)

# fill a matrix with the function values

zz = pylab.zeros(xx.shape)

for i in range(xx.shape[0]):

for j in range(xx.shape[1]):

zz[i,j] = func(xx[i,j], yy[i,j])

# plot the calculated function values

pylab.pcolor(xx,yy,zz)

# and a color bar to show the correspondence between function value and color

pylab.colorbar()

pylab.show()