使用specutils对Python进行连续拟合

import numpy as np
import matplotlib.pyplot as plt
from specutils import Spectrum1D, fitting

wavelength = np.array([5000, 5100, 5200, 5300, 5400])  # 波长数据
flux = np.array([1.0, 1.2, 1.5, 1.3, 1.1])  # 强度数据

spectrum = Spectrum1D(spectral_axis=wavelength, flux=flux)  # 创建光谱对象

model = fitting.models.Polynomial1D(degree=1)  # 一阶多项式模型

fitter = fitting.LevMarLSQFitter()  # 拟合器
continuum = fitter(spectrum.spectral_axis, spectrum.flux, model)  # 连续拟合结果

plt.plot(spectrum.spectral_axis, spectrum.flux, label='原始数据')
plt.plot(spectrum.spectral_axis, continuum(spectrum.spectral_axis), label='连续拟合结果')
plt.xlabel('波长')
plt.ylabel('强度')
plt.legend()
plt.show()