matplotlib 所有热图颜色条显示在最后一个图上

yyyllmsg  于 2023-06-23  发布在  其他
关注(0)|答案(1)|浏览(79)

我在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()
ki0zmccv

ki0zmccv1#

这里的技巧是直接对对象进行操作。因此,使用fig.colorbar代替colorbar
正如你在问题中提到的,from pylab import *是非常不鼓励的。将代码升级到面向对象的接口很简单:

import numpy as np
from matplotlib import pyplot

# 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=pyplot.cm.viridis, interpolation='nearest', 
                   aspect='auto')
    ax.set_ylabel('Y')
    ax.set_xlabel('X')
    cbar = fig.colorbar(im)  # change: use fig.colorbar
    cbar.set_label(r'$Z_{{}}$'.format(subscript))
    fig.suptitle(title)
    fig.tight_layout()

# Plot the heatmaps
## change: use the pyplot function
fig1 = pyplot.figure()
fig2 = pyplot.figure()
fig3 = pyplot.figure()

DoPlot(fig1, Z1, l1)
DoPlot(fig2, Z2, l2)
DoPlot(fig3, Z3, l3)

pyplot.show()  ## change

您还可以使用一个简单的for循环来减少代码的重复,但这超出了本问题的范围。

相关问题