来自xyz数据的Matplotlib轮廓：griddata无效索引

Question

我正在尝试使用以下格式的文件的matplotlib绘制轮廓图：

x1 y1 z1

x2 y2 z2

等等

我可以用numpy.loadtxt加载它以获得向量。到目前为止，没有问题。

我阅读了这篇文章以了解如何进行绘制，并且可以通过复制粘贴来复制它，因此，我确定我的安装没有错：

http://matplotlib.org/examples/pylab_examples/griddata_demo.html

我知道我必须将x和y输入为向量，将z输入为数组，这可以使用griddata完成。这也是我在本网站上找到的。

该文件说：

zi = griddata（x，y，z，xi，yi）使形式z = f （
x，y）的表面适合（通常）非均匀间隔向量（x，y，z）中的数据。griddata（）在（xi，yi）指定的点上插值该曲面以生成zi。xi和yi必须描述规则的网格，可以是1D或2D，但必须单调递增。

为了示例，我编写了以下代码：

import numpy as np
import matplotlib.pyplot as plt
import matplotlib.mlab as ml

x=np.linspace(1.,10.,20)
y=np.linspace(1.,10.,20)
z=np.linspace(1.,2.,20)

xi=np.linspace(1.,10.,10)
yi=np.linspace(1.,10.,10)

zi = ml.griddata(x,y,z,xi,yi)

但是，当涉及到网格数据时，出现以下错误：IndexError：无效索引

因此，我尝试对文档的示例进行如下修改：

from matplotlib.mlab import griddata
import matplotlib.pyplot as plt
import numpy as np
x = np.linspace(-2.1,2.1,300)
y = np.linspace(-2.1,2.1,300)
z = x*np.exp(-x**2-y**2)
# define grid.
xi = np.linspace(-2.1,2.1,100)
yi = np.linspace(-2.1,2.1,200)
# grid the data.
zi = griddata(x,y,z,xi,yi,interp='linear')

而且我得到同样的错误。我不明白怎么了。

谢谢你的帮助。

Answer 1

好的，我终于找到了绘制它的解决方案。对于那些感兴趣的人，这是窍门：将Scipy中的griddata与’nearest’方法一起使用。

from scipy.interpolate import griddata
import numpy as np
import matplotlib.pyplot as plt
x=np.linspace(1.,10.,20)
y=np.linspace(1.,10.,20)
z=z = np.random.random(20)
xi=np.linspace(1.,10.,10)
yi=np.linspace(1.,10.,10)

X,Y= np.meshgrid(xi,yi)
Z = griddata((x, y), z, (X, Y),method='nearest')
plt.contourf(X,Y,Z)