python - 制作散点轮廓

Question

在python中，如果我有一组数据

x, y, z

我可以做一个分散的

import matplotlib.pyplot as plt
plt.scatter(x,y,c=z)

如何获得散点图的 plt.contourf(x,y,z) ？

Answer 1

您可以使用 tricontourf正如 this other answer 的 b. 中所建议的那样:

import matplotlib.tri as tri
import matplotlib.pyplot as plt

plt.tricontour(x, y, z, 15, linewidths=0.5, colors='k')
plt.tricontourf(x, y, z, 15)

旧回复:

使用以下函数转换为contourf需要的格式:

from numpy import linspace, meshgrid
from matplotlib.mlab import griddata

def grid(x, y, z, resX=100, resY=100):
    "Convert 3 column data to matplotlib grid"
    xi = linspace(min(x), max(x), resX)
    yi = linspace(min(y), max(y), resY)
    Z = griddata(x, y, z, xi, yi)
    X, Y = meshgrid(xi, yi)
    return X, Y, Z

现在你可以这样做了:

X, Y, Z = grid(x, y, z)
plt.contourf(X, Y, Z)