感谢matplotlib，如何在同一图中显示具有重叠数据的多个图形？

Question

我正在搜索从同一个图中具有不同轴 X、Y 和数据 Z 的颜色图或轮廓函数绘制多个图形（这里是 2 个）。但是，我只想用一个颜色条为所有图表显示每个数据的最大值。

在此示例中，我创建了一个图，在其中添加了每个图，但第二个图会覆盖第一个图，无论其数据是更低还是更高。

import matplotlib.pyplot as plt
import numpy as np

a = [1,0.25]

fig = plt.figure(1)
ax = fig.gca()

for i in range(2):
    x = np.linspace(-3, 3, 51)
    y = np.linspace(-2*a[i], 2*a[i], 41)

    X, Y = np.meshgrid(x, y)
    if i == 0:
        Z = (1 - X/2 + X**5 + Y**3) * np.exp(-X**2 - Y**2)
    else:
        Z = 0.5*np.ones((41,51))

    graph = ax.contourf(X,Y,Z)  
    bar = fig.colorbar(graph)

plt.show()

Figure 1 displayed by the code

这是我要显示的内容：

Figure 2 desired

非常感谢， 特里斯坦

Answer 1

根据我们在 cmets 中对您帖子的讨论，我认为您可以编辑您的代码以实现您想要的效果，如下所示。

首先，作为一般性评论，我建议您将变量移到脚本的顶部。

其次，这是主要部分，如果您使用比较来测试要填充 Z 数组的值，您可以只绘制一个图表。您可以使用 np.logical_and 链接多个比较，然后使用 np.where 根据您是否在所需的 x 和 y 值框内以​​及是否使用函数值或常数值填充 Z 数组函数值或期望的常数值最大。

fig = plt.figure()
ax = fig.gca()

xmin, xmax, nx = -3, 3, 51
ymin, ymax, ny = -2, 2, 41

# box values
xbmin, xbmax = -3, 3
ybmin, ybmax = -0.5, 0.5
zlevel = 0.5

x = np.linspace(xmin, xmax, nx)
y = np.linspace(ymin, ymax, ny)
X, Y = np.meshgrid(x,y)
Z = (1 - X/2 + X**5 + Y**3) * np.exp(-X**2 - Y**2)

test1 = Z<zlevel
test2 = np.logical_and(X>=xbmin, X<=xbmax)
test3 = np.logical_and(Y>=ybmin, Y<=ybmax)

mask = np.logical_and(np.logical_and(test1, test2), test3)
Z = np.where(mask, zlevel*np.ones(Z.shape), Z)

graph = ax.contourf(X,Y,Z)
bar = fig.colorbar(graph)

plt.show()