輪廓和輪廓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()

但我有這個例外：

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必須是二維數組。

如果您的數據不在網格上，您要么需要將其插入到網格中，要么不能使用等高線。

一個簡單的替代方法是使用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 。

您還可以查看：

• 在 matplotlib 中從 (x, y, z) 數據集繪制等值線/等高線（對於三角和插值等高線都有一個簡單的例子）
• **concave** 非網格數據的 matplotlib 輪廓/輪廓（對於你的數據是凹的）