mirror of
https://github.com/Findus23/collision-analyisis-and-interpolation.git
synced 2024-09-19 15:13:50 +02:00
12 lines
423 B
Python
12 lines
423 B
Python
from numpy import ndarray, save
|
|
from scipy.interpolate import griddata
|
|
|
|
from interpolators.base import BaseInterpolator
|
|
|
|
|
|
class GriddataInterpolator(BaseInterpolator):
|
|
def __init__(self, *args, **kwargs):
|
|
super().__init__(*args, **kwargs)
|
|
|
|
def interpolate(self, alpha, v, mcode, gamma, wt, wp) -> ndarray:
|
|
return griddata(self.points, self.values, (alpha, v, mcode, gamma, wt, wp), method="linear")
|