Python开发之手动实现一维线性插值
- 1.线性插值法介绍
- 2.手动实现线性插值
- 3.案例一:手动实现线性插值
- 4.使用pandas的插值方法实现要求(推荐)
- 5.案例二:对一组数据进行线性插值和SG滤波处理
前言:主要介绍手动实现一维线性插值以及pandas里面的interpolate方法实现线性插值。并结合案例实现一组数据的线性插值和SG滤波处理。
1.线性插值法介绍
- 线性插值法是一种简单的插值方法,用于在两个已知数据点之间估算未知位置上的数据值。它基于线性关系的假设,即在两个已知数据点之间的数据值随着位置的变化而呈线性变化。
- 假设有两个已知数据点 (x1, y1) 和 (x2, y2),其中 x1 和 x2 是两个数据点的位置,y1 和 y2 是对应的数据值。现在需要在这两个数据点之间的某个位置 x0 处估算数据值 y0。
- 线性插值法的计算公式如下:
y0 = y1 + (x0 - x1) * (y2 - y1) / (x2 - x1)
其中,y0 是在位置 x0 处估算的数据值,y1 和 y2 是已知数据点的数据值,x1 和 x2 是已知数据点的位置,x0 是要估算的位置。
线性插值法的原理是利用已知数据点之间的线性关系,根据位置的比例关系来估算未知位置上的数据值。它假设数据值在两个已知数据点之间是线性变化的,因此在两个数据点之间的位置上,估算值与实际值的误差较小。 - 线性插值法在实际应用中常用于处理缺失数据或填补异常值,以及在图形绘制和数据处理中用于平滑数据。然而,需要注意的是,线性插值法在数据变化非线性或存在较大波动时可能不适用,此时需要考虑其他更复杂的插值方法。
2.手动实现线性插值
核心代码
def linear_interpolation(x1, y1, x2, y2, x0):"""线性插值函数参数:x1, y1: 第一个已知数据点的位置和数据值x2, y2: 第二个已知数据点的位置和数据值x0: 要估算的位置返回值:y0: 在位置 x0 处估算的数据值"""y0 = y1 + (x0 - x1) * (y2 - y1) / (x2 - x1)return y0# 示例数据
x1, y1 = 0, 10
x2, y2 = 5, 20
x0 = 2# 使用线性插值估算在位置 x0 处的数据值
y0 = linear_interpolation(x1, y1, x2, y2, x0)
print(f"在位置 {x0} 处的估算数据值为:{y0}")
3.案例一:手动实现线性插值
数据格式:
分析:1.首先判断该数据是否需要插值
2.获取nan的索引位置,保存到list集合中去
3.根据获取的nan的索引位置,得到位置前后的数据,通过线性插值法算出该nan的值。
代码具体实现:
import numpy as np
import pandas as pddef is_exist_nan(data):for d in data:if np.isnan(d):return Truereturn Falsedef get_nan_index_list(data):index_list = []for index,d in enumerate(data):if np.isnan(d):index_list.append(index)return index_listdef linear_interpolation(x1, y1, x2, y2, x0):"""线性插值函数参数:x1, y1: 第一个已知数据点的位置和数据值x2, y2: 第二个已知数据点的位置和数据值x0: 要估算的位置返回值:y0: 在位置 x0 处估算的数据值"""y0 = y1 + (x0 - x1) * (y2 - y1) / (x2 - x1)return y0def get_first_data(data):for index,d in enumerate(data):if not np.isnan(d):return ddef get_last_data(data):count = len(data) -1for d in data:if not np.isnan(data[count]):return data[count]else:count=count-1def digu(x2,data):if not np.isnan(data[x2]):return x2,dataelse:x2 = x2 + 1return digu(x2,data)def get_new_data(nan_index_list, data):if nan_index_list[0] == 0:data[0] = get_first_data(data)nan_index_list.remove(0)if len(nan_index_list)>=1:if nan_index_list[len(nan_index_list)-1] == 26:data[26] =get_last_data(data)nan_index_list.remove(26)if len(nan_index_list) >=1:for nan_index in nan_index_list:x1 = nan_index - 1y1 = data[x1]x2 = nan_index + 1x2,data = digu(x2,data)y2 = data[x2]x0 = nan_indexy0 = round(linear_interpolation(x1, y1, x2, y2, x0), 4)data[nan_index] = y0return dataif __name__ == '__main__':data1 = [np.nan, -0.3356, -0.3208, -0.3661, 0.2192, np.nan, np.nan, np.nan, -0.3709, -0.3779, 0.026, -0.2601,np.nan, -0.0238, -0.2241, -0.2105, -0.2623, 0.379, -0.2196, np.nan, -0.0835, 0.2895, 0.0415, -0.2323,-0.1782, -0.2308, -0.2265]if is_exist_nan(data1):print(data1)nan_index_list = get_nan_index_list(data1)new_data = get_new_data(nan_index_list,data1)print(new_data)
运行结果如下:
[nan, -0.3356, -0.3208, -0.3661, 0.2192, nan, nan, nan, -0.3709, -0.3779, 0.026, -0.2601, nan, -0.0238, -0.2241, -0.2105, -0.2623, 0.379, -0.2196, nan, -0.0835, 0.2895, 0.0415, -0.2323, -0.1782, -0.2308, -0.2265]
[-0.3356, -0.3356, -0.3208, -0.3661, 0.2192, 0.0717, -0.0758, -0.2234, -0.3709, -0.3779, 0.026, -0.2601, -0.1419, -0.0238, -0.2241, -0.2105, -0.2623, 0.379, -0.2196, -0.1515, -0.0835, 0.2895, 0.0415, -0.2323, -0.1782, -0.2308, -0.2265]
给大家提供一个思路,具体用的时候,推荐用pandas的interpolate方法实现。
4.使用pandas的插值方法实现要求(推荐)
import pandas as pdif __name__ == '__main__':# 原始数据,包含缺失值data = [np.nan, -0.3356, -0.3208, -0.3661, 0.2192, np.nan, np.nan, np.nan, -0.3709, -0.3779, 0.026, -0.2601,np.nan, -0.0238, -0.2241, -0.2105, -0.2623, 0.379, -0.2196, np.nan, -0.0835, 0.2895, 0.0415, -0.2323,-0.1782, -0.2308, -0.2265]# 将数据转换为pandas的Series对象,此时缺失值会自动转换为NaNdata_series = pd.Series(data)# 执行线性插值,并处理第一个和最后一个NaNinterpolated_data = data_series.interpolate(limit_direction='both')# 打印插值结果print(interpolated_data.values)
5.案例二:对一组数据进行线性插值和SG滤波处理
import numpy as np
import pandas as pd
from scipy.signal import savgol_filter
import matplotlib.pyplot as pltdef get_interpolated_data(data):# 将数据转换为pandas的Series对象,此时缺失值会自动转换为NaNdata_series = pd.Series(data)# 执行线性插值,并处理第一个和最后一个NaNinterpolated_data = data_series.interpolate(limit_direction='both').tolist()interpolated_data = [round(i, 4) for i in interpolated_data] # 保留四位小数return interpolated_datadef get_sg_data(data, window_size, polyorder):smoothed_data = savgol_filter(data, window_size, polyorder).tolist()smoothed_data = [round(i, 4) for i in smoothed_data] # 保留四位小数return smoothed_dataif __name__ == '__main__':# 原始数据,包含缺失值data = [np.nan, -0.3356, -0.3208, -0.3661, 0.2192, np.nan, np.nan, np.nan, -0.3709, -0.3779, 0.026, -0.2601,np.nan, -0.0238, -0.2241, -0.2105, -0.2623, 0.379, -0.2196, np.nan, -0.0835, 0.2895, 0.0415, -0.2323,-0.1782, -0.2308, -0.2265]interpolated_data = get_interpolated_data(data)sg_data = get_sg_data(interpolated_data,5,2)sg_data2 = get_sg_data(interpolated_data,9,3)print(interpolated_data)print(sg_data)print(sg_data2)plt.plot(interpolated_data, label='interpolated_data')plt.plot(sg_data, label='sg_data window_size=5 polyorder=2')plt.plot(sg_data2, label='sg_data2 window_size=9 polyorder=3')plt.xlabel('Time')plt.ylabel('Value')plt.title('Line Plot')plt.legend()plt.show()