在Python中将Colorbar和Contour Map添加到PyPlot表面imshow

Question

我使用的是前雇员编写的代码。在此代码中，该人使用Qt在Python上执行绘图。我正在尝试添加颜色条和等高线图，但无济于事。这是相关代码的一部分。

#IMPORTS
from matplotlib.figure import Figure
from matplotlib.backends.backend_qt4agg import FigureCanvasQTAgg as FigureCanvas

#SETUP
fig = Figure()
axes = fig.add_subplot(111)
canvas = FigureCanvas(fig)

#DRAWING
axes.clear()
axes.imshow(data_and_extents, interpolation='bilinear')
axes.autoscale(False)
canvas.draw()

#TODO: Add colorbar and contour maps with some values

我尝试过axes.colorbar（）无济于事。如何将颜色条和等高线图添加到这样设置的曲面图中？

感谢SOCommunity！

Answer 1

第一个错误出现在第一行：

from matplotlib.figure import Figure

是正确的。 下一个：而不是axes.colorbar它是fig.corbar。 使用变量名称，您应该能够创建一个颜色栏：

cbar = fig.colorbar(axes)

有关详情，请参阅this example。