Question

python菜鸟求助，使用matplotlib 绘制contour等高线图，z为2D数组

Answer 1

以画 z=x^2+y^2 的等高线为例，简单介绍用 matplotlib 画等高线的方法.

首先看下 z=x^2+y^2 的三维图像：

import numpy as npfrom matplotlib import cmimport matplotlib.pyplot as pltfrom mpl_toolkits.mplot3d import Axes3D

delta = 0.2x = np.arange(-3, 3, delta)

y = np.arange(-3, 3, delta)

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

Z = X**2 + Y**2x=X.flatten()

y=Y.flatten()

z=Z.flatten()

fig = plt.figure()

ax = fig.gca(projection='3d')

ax.plot_trisurf(x, y, z, cmap=cm.jet, linewidth=0.01)

plt.show()