matplotlib中的距离依赖着色

Question

我想创建一些电磁散射过程的远场图。

为此，我计算了θ，φ和r值。坐标θ和φ在unitphere上创建一个规则网格，所以我可以使用plot_Surface（found here）转换为笛卡尔坐标。

我现在的问题是，我需要一种方法来对半径为r而不是高度为z的表面着色，这似乎是默认的。

有没有办法改变这种依赖？

Answer 1

我不知道你是怎么做的，所以也许你已经解决了它。但是，基于Paul评论的链接，你可以做这样的事情。我们使用plot_surface的facecolor参数传递我们想要的颜色值。

（我已经从matplotlib文档修改的surface3d演示）

编辑：正如斯蒂芬在他的评论中指出的那样，我的回答可以简化为：

from mpl_toolkits.mplot3d import Axes3D 
import matplotlib.pyplot as plt 
import matplotlib.cm as cm 
import numpy as np 

fig = plt.figure() 
ax = fig.gca(projection='3d') 
X = np.arange(-5, 5, 0.25) 
xlen = len(X) 
Y = np.arange(-5, 5, 0.25) 
ylen = len(Y) 
X, Y = np.meshgrid(X, Y) 
R = np.sqrt(X**2 + Y**2) 
maxR = np.amax(R) 
Z = np.sin(R) 

# Note that the R values must still be normalized. 
surf = ax.plot_surface(X, Y, Z, rstride=1, cstride=1, facecolors=cm.jet(R/maxR), 
     linewidth=0) 

plt.show() 

和（结束）我不必要的复杂的原始版本，使用与上面相同的代码，但省略matplotlib.cm导入，

# We will store (R, G, B, alpha) 
colorshape = R.shape + (4,) 
colors = np.empty(colorshape) 
for y in range(ylen): 
    for x in range(xlen): 
     # Normalize the radial value. 
     # 'jet' could be any of the built-in colormaps (or your own). 
     colors[x, y] = plt.cm.jet(R[x, y]/maxR) 

surf = ax.plot_surface(X, Y, Z, rstride=1, cstride=1, facecolors=colors, 
     linewidth=0) 

plt.show()