我在Matplotlib(Python 3.6.0)中绘制多个热图时遇到了一个问题。
我有一个函数可以绘制一些数据的热图,每个热图都在一个单独的图中。当我对不同的数据数组运行此函数时,热图都在各自的图中显示得很好,但由于某种原因,它们的所有颜色条都显示在最近绘制的热图的图上,如下面链接的图像所示。
热图错误
请注意,当我尝试在不使用该函数的情况下手动绘制热图时,此行为仍然存在。还要注意的是,颜色条不会简单地显示在最近绘制的图上,而是仅显示在包含热图的最近绘制的图上。例如,如果我稍后绘制一个线图,则颜色条不会显示在该线图上,而只会显示在最后一个热图上。
下面是一个最小的工作示例:
import numpy as np
from pylab import *
# Function
f1 = lambda X, Y: X*X + Y*Y
f2 = lambda X, Y: X*X - Y*Y
f3 = lambda X, Y: X*Y - Y
# Grid on which function is to be evaluated
x = np.arange(0, 100, 1)
y = np.arange(0, 100, 1)
Xaxis = x[:, None]
Yaxis = y[None, :]
# Evaluate functions and create labels for plotting
Z1 = f1(Xaxis, Yaxis)
l1 = ['F1', '1']
Z2 = f2(Xaxis, Yaxis)
l2 = ['F2', '2']
Z3 = f3(Xaxis, Yaxis)
l3 = ['F3', '3']
# Function to plot heatmaps
def DoPlot(fig, fun, label):
title = label[0]
subscript = label[1]
ax = fig.add_subplot(111)
im = ax.imshow(fun, cmap=cm.viridis, interpolation='nearest',
aspect='auto')
ax.set_ylabel('Y')
ax.set_xlabel('X')
cbar = colorbar(im)
cbar.set_label(r'$Z_{{}}$'.format(subscript))
fig.suptitle(title)
fig.tight_layout()
# Plot the heatmaps
fig1 = figure()
fig2 = figure()
fig3 = figure()
DoPlot(fig1, Z1, l1)
DoPlot(fig2, Z2, l2)
DoPlot(fig3, Z3, l3)
show()
1条答案
按热度按时间ki0zmccv1#
这里的技巧是直接对对象进行操作。因此,使用
fig.colorbar
代替colorbar
。正如你在问题中提到的,
from pylab import *
是非常不鼓励的。将代码升级到面向对象的接口很简单:您还可以使用一个简单的
for
循环来减少代码的重复,但这超出了本问题的范围。