Matplotlib - 单值等高线图

Question

我想绘制一些数据的等值线图，但字段中的所有值可能都相同。这会导致 matplotlib 中出现错误，这是有道理的，因为实际上没有要创建的轮廓。例如，如果运行下面的代码，您将收到错误，但删除 zi 的第二个定义，它会按预期运行。

如果某些数据是均匀场，如何制作“等高线”图？我希望它看起来就像常规等高线图（有一个填充某种颜色的盒子并在侧面显示颜色条。颜色条可以是统一的颜色，或者仍然显示 15 种颜色的范围，我不知道不在乎）。

代码：

from numpy        import array
import matplotlib.pyplot as plt

xi = array([0., 0.5, 1.0])
yi = array([0., 0.5, 1.0])
zi = array([[0., 1.0, 2.0],
            [0., 1.0, 2.0],
            [0., 1.0, 2.0]])
zi = array([[1.0, 1.0, 1.0],
            [1.0, 1.0, 1.0],
            [1.0, 1.0, 1.0]])

CS = plt.contour(xi, yi, zi, 15, linewidths=0.5, colors='k')
CS = plt.contourf(xi, yi, zi, 15, cmap=plt.cm.jet)
plt.colorbar()
plt.show()

Answer 1

好吧，contourf 完美地处理了它，令人窒息的是 contour。

为什么不这样做：

import numpy as np
import matplotlib.pyplot as plt

xi = np.array([0., 0.5, 1.0])
yi = np.array([0., 0.5, 1.0])
zi = np.ones((3,3))

try:
    CS = plt.contour(xi, yi, zi, 15, linewidths=0.5, colors='k')
except ValueError:
    pass
CS = plt.contourf(xi, yi, zi, 15, cmap=plt.cm.jet)

plt.colorbar()
plt.show()

这样，如果存在均匀场，您将得到一个填充的（默认为绿色）框，否则将得到一个带有线条的填充等值线图。