matplotlib - 在pcolor plot python中绘制多组数据

我有像(x，y，(z1，z2，z3 ..))的数据集。我在尝试

plt.pcolor(x,y,z1)
plt.pcolor(x,y,z2)
plt.pcolor(x,y,z3)
plt.colorbar()
plt.show()

这仅显示了最后一个数据集的pcolor图。如何在相同的图和相同的颜色条比例下绘制所有图？

最佳答案

您可以尝试使用subplots，并确保所有具有相同强度比例的图像(对所有图像使用相同的vmin和vmax参数pcolor())。下面是一个示例:

import numpy as np
import matplotlib.pyplot as plt

dx, dy = 0.15, 0.05

y, x = np.mgrid[slice(-3, 3 + dy, dy),
                slice(-3, 3 + dx, dx)]
z = (1 - x / 2. + x ** 5 + y ** 3) * np.exp(-x ** 2 - y ** 2)
z1 = z[:-1, :-1]
z2 = z[:-1, :-1]
z3 = z[:-1, :-1]
z_min, z_max = -np.abs(z).max(), np.abs(z).max()

data = [[x,y,z1],[x,y,z2],[x,y,z3]]

# Plot each slice as an independent subplot
fig, axes = plt.subplots(nrows=1, ncols=3)
for dat, ax in zip(data, axes.flat):
    # The vmin and vmax arguments specify the color limits
    pc = ax.pcolor(dat[0],dat[1],dat[2], vmin=z_min, vmax=z_max)
# Make an axis for the colorbar on the right side
cax = fig.add_axes([0.9, 0.1, 0.03, 0.8])
fig.colorbar(pc, cax=cax)

plt.show()

它将显示如下:

关于matplotlib - 在pcolor plot python中绘制多组数据，我们在Stack Overflow上找到一个类似的问题：https://stackoverflow.com/questions/43963438/