Linear interpolation in Python using interpolate.interp2d