[英]Colorbar for matplotlib plot_surface command
I have modified the mplot3d example code for my application with Paul's help.在 Paul 的帮助下,我为我的应用程序修改了 mplot3d示例代码。 The code reads:
代码如下:
from mpl_toolkits.mplot3d import Axes3D
import matplotlib.pyplot as plt
import numpy as np
fig = plt.figure()
ax = fig.add_subplot(111, projection='3d')
u = np.linspace(0, 2 * np.pi, 100)
v = np.linspace(0, np.pi, 100)
x = 10 * np.outer(np.cos(u), np.sin(v))
y = 10 * np.outer(np.sin(u), np.sin(v))
z = 10 * np.outer(np.ones(np.size(u)), np.cos(v))
z1 = z * np.cos(0.5*x)
N = z1 / z1.max() # normalize 0..1
surf = ax.plot_surface(x, y, z, rstride=1, cstride=1, facecolors=cm.jet(N), linewidth=0, antialiased=False, shade=False)
fig.colorbar(surf, shrink=0.5, aspect=5)
plt.show()
The code works great to plot the surface.该代码适用于 plot 表面。 But when I try to add a colorbar to the above plot, I get the following error:
但是当我尝试在上面的 plot 中添加颜色条时,出现以下错误:
Traceback (most recent call last):
File "<ipython console>", line 1, in <module>
File "C:\Python26\lib\site-packages\spyderlib\widgets\externalshell\startup.py", line 122, in runfile
execfile(filename, glbs)
File "C:\Documents and Settings\mramacha\My Documents\Python\Candela\test.py", line 22, in <module>
fig.colorbar(surf, shrink=0.5, aspect=5)
File "C:\Python26\lib\site-packages\matplotlib\figure.py", line 1104, in colorbar
cb = cbar.Colorbar(cax, mappable, **kw)
File "C:\Python26\lib\site-packages\matplotlib\colorbar.py", line 706, in __init__
mappable.autoscale_None() # Ensure mappable.norm.vmin, vmax
File "C:\Python26\lib\site-packages\matplotlib\cm.py", line 261, in autoscale_None
raise TypeError('You must first set_array for mappable')
TypeError: You must first set_array for mappable
I would be so grateful if someone is able to guide me with this.如果有人能够指导我,我将不胜感激。
Prabhu帕布
You can use a proxy mappable object since your surface array is not mapped.您可以使用代理可映射 object,因为您的表面阵列未映射。 The mappable simply converts the values of any array to RGB colors defined by a colormap.
可映射对象只是将任何数组的值转换为由颜色图定义的 RGB colors。
In your case you want to do this with the z1 array:在您的情况下,您想使用 z1 数组执行此操作:
import matplotlib.cm as cm
m = cm.ScalarMappable(cmap=cm.jet)
m.set_array(z1)
plt.colorbar(m)
声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.