在 pcolormesh 上叠加轮廓颜色条

Question

我有以下 python 代码：

import numpy as np

import matplotlib.pyplot as plt

x = np.linspace(0, 10, 100)

y = np.linspace(0, 10, 100)

X, Y = np.meshgrid(x, y, indexing='ij')

Z = np.sin(X+Y)

plt.pcolormesh(X, Y, Z)

plt.colorbar()

plt.contour(X, Y, Z, levels=[0.5, 0.75], colors=['black','cyan'])

plt.show()

给出以下输出：

我想将轮廓标记放置在颜色条上，如下所示：

我已经进行了实验和阅读，但我找不到一种方法可以有效地在同一颜色条上绘制两个数量。

Answer 1

plt.colorbar返回一个colorbar对象，您可以从中获取它使用 绘制的轴.ax。从那里开始，事情应该很简单：plt.pcolormesh(X, Y, Z)cb = plt.colorbar()plt.contour(X, Y, Z, levels=[0.5, 0.75], colors=['black','cyan'])ax = cb.axxmin, xmax = ax.get_xlim()ax.hlines([0.5, 0.75], xmin, xmax, colors=['black','cyan'], linewidth=5)输出：