python三维数据增强_如何用SciPy提高三维数据插值的性能

我有代表大气的三维数据。现在我想把这个数据插值到一个公共的Z坐标(我的意思应该从函数的doctring中明确)。下面的代码可以正常工作，但是我想知道是否有一种方法可以提高性能。。。在def interpLevel(grid,value,data,interp='linear'):

"""

Interpolate 3d data to a common z coordinate.

Can be used to calculate the wind/pv/whatsoever values for a common

potential temperature / pressure level.

grid : numpy.ndarray

The grid. For example the potential temperature values for the whole 3d

grid.

value : float

The common value in the grid, to which the data shall be interpolated.

For example, 350.0

data : numpy.ndarray

The data which shall be interpolated. For example, the PV values for

the whole 3d grid.

kind : str

This indicates which kind of interpolation will be done. It is directly

passed on to scipy.interpolate.interp1d().

returs : numpy.ndarray

A 2d array containing the *data* values at *value*.

"""

ret = np.zeros_like(data[0,:,:])

# we need to copy the grid to a new one, because otherwise the flipping

# done below will be messed up

gr = np.zeros_like(grid)

da = np.zeros_like(data)

for latIdx in xrange(grid.shape[1]):

for lonIdx in xrange(grid.shape[2]):

# check if we need to flip the column

if grid[0,latIdx,lonIdx] > grid[-1,latIdx,lonIdx]:

gr[:,latIdx,lonIdx] = grid[::-1,latIdx,lonIdx]

da[:,latIdx,lonIdx] = data[::-1,latIdx,lonIdx]

else:

gr[:,latIdx,lonIdx] = grid[:,latIdx,lonIdx]

da[:,latIdx,lonIdx] = data[:,latIdx,lonIdx]

f = interpolate.interp1d(gr[:,latIdx,lonIdx], \

da[:,latIdx,lonIdx], \

kind=interp)

ret[latIdx,lonIdx] = f(value)

return ret