python中的等高线图

【问题标题】：Contour graph in pythonpython中的等高线图
【发布时间】：2013-03-24 16:39:36
【问题描述】：

如何使用matplotlib.pyplot 在python 中创建计数网格，其中网格是z 变量低于零的一种颜色，而z 等于或大于零时的另一种颜色？我对matplotlib 不是很熟悉，所以如果有人能给我一个简单的方法，那就太好了。

到目前为止我有：

x= np.arange(0,361)
y= np.arange(0,91)

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

area = funcarea(L,D,H,W,X,Y) #L,D,H and W are all constants defined elsewhere.

plt.figure()
plt.contourf(X,Y,area)
plt.show()

【问题讨论】：

你使用的是哪个版本的python（2或3）

标签： python matplotlib contour

【解决方案1】：

您可以使用 contourf 中的 levels 关键字来执行此操作。

import numpy as np
import matplotlib.pyplot as plt

fig, axs = plt.subplots(1,2)

x = np.linspace(0, 1, 100)
X, Y = np.meshgrid(x, x)
Z = np.sin(X)*np.sin(Y)

levels = np.linspace(-1, 1, 40)

zdata = np.sin(8*X)*np.sin(8*Y)

cs = axs[0].contourf(X, Y, zdata, levels=levels)
fig.colorbar(cs, ax=axs[0], format="%.2f")

cs = axs[1].contourf(X, Y, zdata, levels=[-1,0,1])
fig.colorbar(cs, ax=axs[1])

plt.show()

您可以通过选择不同的colormap来改变颜色；使用 vmin、vmax；等等

【讨论】：