使用 dataframe 的最近点在 dataframe 的点之间进行线性插值

Question

如果有大量数据——数百万个点（没有 NAN），我需要一个快速的解决方案来在数据帧的最近点之间进行插值，而不向数据帧添加新点。 dataframe 使用 x 值排序。

例如，我有一个带有下一列的 dataframe：

x | y
-----
0 | 1
1 | 2
2 | 3
...

我需要一个 function，它将针对给定的 x_input 值触发，该值计算为最近点之间的线性插值，如下所示：

calc_linear(df, cinput_col = 'x', input_val=1.5, output_col=y)将 output 2.5 - 作为给定 x 的插值 y 值

也许有一些 pandas 功能？

Answer 1

使用numpy.interp ：

import numpy as np

def calc_linear(df, input_val, input_col='x', output_col='y'):
    return np.interp(input_val, df[input_col], df[output_col])

y = calc_linear(df, 1.5)
print(y)

# Output
2.5