[SciPy-User] interpolation with inverse-distance weighting + KDTree

denis denis-bz-gg@t-online...
Wed Jun 30 05:06:45 CDT 2010


Folks,
  here's a tiny class Invdisttree for interpolation with inverse-
distance weighting + KDTree.
It's solid, pretty fast, local, works for scattered data in any number
of dimensions,
leverages the excellent KDTree module.
Comments would be welcome; real test cases, 3+d, most welcome.

(For interpolating 2d data to a fine uniform grid,
matplotlib._delaunay.nn_interpolate_grid is ~ 10 times faster (on my
old mac ppc
One reason is that the dot( 1/dist, z[ix] ) takes over half the time;
another may be that nn_grid caches the current triangle ? )

cheers
  -- denis


    import numpy as np
    from scipy.spatial import cKDTree as KDTree

    class Invdisttree:
        """ inverse-distance-weighted interpolation using KDTree:
        invdisttree = Invdisttree( X, z )  -- points, values
        interpol = invdisttree( q, k=6, eps=0 )
            -- interpolate z from the 6 points nearest each q;
               q may be one point, or a batch of points
        """
        def __init__( self, X, z, leafsize=10 ):
            self.tree = KDTree( X, leafsize=leafsize )  # build the
tree
            self.z = z

        def __call__( self, q, k=6, eps=0 ):
                # k nearest neighbours of each query point --
            self.distances, self.ix = self.tree.query( q, k=k,
eps=eps )
            interpol = []  # np.zeros( (len(self.distances),) +
np.shape(z[0]) )
            for dist, ix in zip( self.distances, self.ix ):
                if dist[0] > 1e-10:
                    w = 1 / dist
                    wz = np.dot( w, self.z[ix] ) / np.sum(w)  # weight
z s by 1/dist
                else:
                    wz = self.z[ix[0]]
                interpol.append( wz )
            return interpol


More information about the SciPy-User mailing list