python - Matplotlib:等高线水平作为颜色条中的线

Question

我正在使用 Matplotlib 绘制一些二维数据作为 pcolor()，然后将其与 contour() 叠加。

当我使用 colorbar() 时，我得到以下颜色条之一:

如何使等高线级别的水平线(左)也显示在彩色条(右)中？

Answer 1

根据您修改后的问题，我明白您的意思。这仍然可以使用 add_lines 来完成。此函数将非填充等高线图中的线条添加到颜色条中。可以找到文档 here .

因此，通过首先根据您的 pcolor 绘图定义颜色条，您可以稍后将 contour 中的线添加到该颜色条:

import numpy
import matplotlib.pyplot as plt
import matplotlib.mlab as mlab

#Generate data
delta = 0.025

x = numpy.arange(-3.0, 3.0, delta)
y = numpy.arange(-2.0, 2.0, delta)

X, Y = numpy.meshgrid(x, y)

Z1 = mlab.bivariate_normal(X, Y, 1.0, 1.0, 0.0, 0.0)
Z2 = mlab.bivariate_normal(X, Y, 1.5, 0.5, 1, 1)
# difference of Gaussians
Z = 10.0 * (Z2 - Z1)

#Plot
fig = plt.figure()
ax1 = fig.add_subplot(1, 1, 1)

PC = ax1.pcolor(X, Y, Z)
CF = ax1.contour(X, Y, Z, 50, colors = "black")

cbar = plt.colorbar(PC)
cbar.add_lines(CF)

plt.show()