如何使用scipy.interpolate中的网格数据

Question

我有一个包含一百万行的三列（x像素、y像素、z值）数据。数据来自图像，并且存在重复的z值。现在我需要做一个曲面图。这张照片就是一个完美的例子。但现在输出图像为空。有人能查一下代码吗？

import numpy as np
from enthought.mayavi import mlab
from scipy.interpolate import griddata
x,y,z = np.loadtxt('test.csv',delimiter=',',usecols=(0,1,2),unpack=True) 
xi,yi = np.mgrid[0:3000:3000j, 0:3000:3000j]
zi = griddata((x, y), z, (xi, yi),method='linear')
mlab.surf(xi,yi,zi)
mlab.show()

Answer 1

我无法在没有数据的情况下检查代码，但我怀疑问题在于您使用默认的fill_value=nan作为griddata参数，因此如果您的网格点超出了（x，y）点的空间，则网格中存在NaN，mlab可能无法处理（matplotlib不容易）。尝试设置fill_value=0或其他合适的实数。