python - 轮廓和轮廓f的使用

Question

我有一个要点列表:

pointList = [ [x1,y1,z1], [x2,y2,z2], ... [xn,yn,zn]]

我想绘制这组点的等高线图。

我尝试:

import matplotlib.pyplot as plt
pointList = [ [x1,y1,z1], [x2,y2,z2], ... [xn,yn,zn]]
x = [el[0] for el in pointList]
y = [el[1] for el in pointList]
z = [el[2] for el in pointList]
plt.contourf(x,y,z)
plt.show()

但我有一个异常(exception):

TypeError: Input z must be a 2D array.

这很奇怪，因为在 matplotlib 的文档中我发现:

Call signatures:
contour(Z)
make a contour plot of an array Z. The level values are chosen automatically.
contour(X,Y,Z)
X, Y specify the (x, y) coordinates of the surface

所以我不明白为什么它会失败......

Answer 1

无论是 contour(Z) 或 contour(X,Y,Z)，输入 Z 都必须是 2D数组。

如果您的数据不在网格上，您要么需要将其插值到网格中，要么无法使用等值线。

一个简单的替代方法是使用 tricontour 。

import matplotlib.pyplot as plt
import numpy as np
pointList = [ [x1,y1,z1], [x2,y2,z2], ... [xn,yn,zn]]
pointList = np.array(pointList)
plt.tricontour(pointList[:,0],pointList[:,1],pointList[:,2])
plt.show()

有一个很好的示例，将 tricontour 与插值数据的 contour 进行比较:tricontour_vs_griddata .

您还可以查看:

• Plotting Isolines/contours in matplotlib from (x, y, z) data set (其中有一个三角轮廓和插值轮廓的简单示例)
• matplotlib contour/contourf of **concave** non-gridded data (对于您的数据是凹的情况)