用python写一个光谱基线校正函数

你好，我是C知道。关于你的问题，我可以回答。以下是一个用Python编写的光谱基线校正函数的示例代码：

import numpy as np
from scipy.signal import savgol_filter

def baseline_correction(spectrum, window_size=11, poly_order=3):
    """
    光谱基线校正函数
    :param spectrum: 待校正的光谱数据，一维数组
    :param window_size: Savitzky-Golay滤波器的窗口大小，默认为11
    :param poly_order: Savitzky-Golay滤波器的多项式阶数，默认为3
    :return: 校正后的光谱数据，一维数组
    """
    # 使用Savitzky-Golay滤波器平滑光谱数据
    smoothed_spectrum = savgol_filter(spectrum, window_size, poly_order)
    # 计算光谱数据与平滑后的数据之间的差值
    baseline = spectrum - smoothed_spectrum
    # 将差值取绝对值，然后平滑一下
    baseline = savgol_filter(np.abs(baseline), window_size, poly_order)
    # 将平滑后的差值加回到平滑后的光谱数据上，得到校正后的光谱数据
    corrected_spectrum = smoothed_spectrum + baseline
    return corrected_spectrum

希望这个函数能够帮助到你。如果你有任何问题，请随时问我。