[Numpy-discussion] an np.arange for arrays?

[David Warde-Farley]

On 10-Jul-09, at 1:25 AM, Chris Colbert wrote:

> actually what would be better is if i can pass two 1d arrays X and Y
> both size Nx1
> and get back a 2d array of size NxM where the [n,:] row is the linear
> interpolation of X[n] to Y[n]

This could be more efficient, but here's a solution using mgrid and  
broadcasting:

def interp_grid_thingie(X, Y, M):
	frac = np.mgrid[0:1:M*1j]
	return (1 - frac[np.newaxis, :]) * X[:, np.newaxis] +  
frac[np.newaxis, :] * Y[:, np.newaxis]

In [11]: interp_grid_thingie(arange(1,6), arange(0,5), 5)
Out[11]:
array([[ 1.  ,  0.75,  0.5 ,  0.25,  0.  ],
        [ 2.  ,  1.75,  1.5 ,  1.25,  1.  ],
        [ 3.  ,  2.75,  2.5 ,  2.25,  2.  ],
        [ 4.  ,  3.75,  3.5 ,  3.25,  3.  ],
        [ 5.  ,  4.75,  4.5 ,  4.25,  4.  ]])

David