正弦波拟合

news2024/11/27 12:48:50

正弦波拟合是一种常见的数学方法，用于确定最佳匹配给定数据集的正弦波形。这可以用于各种应用，如信号处理、周期性数据分析等。以下举例展示如何进行正弦波拟合。

步骤与方法

收集数据：首先，你需要收集或生成一组数据，这组数据应该展示出一种周期性变化，类似于正弦波。

选择拟合模型：一般的正弦波形式为 y = A * sin(B * x + C) + D，其中：
A 是振幅（波的高度），
B 决定波的频率，
C 是相位偏移（波的水平移动），
D 是垂直偏移。
使用优化方法：通过调整上述参数，使得拟合的正弦曲线尽可能接近实际数据点。这通常通过最小化误差函数（如最小二乘法）来实现。
分析结果：分析拟合后的参数，确定它们是否符合预期，并检查拟合的质量。

模拟数据进行举例

例1

import numpy as np
import matplotlib.pyplot as plt
from scipy.optimize import curve_fit

# 示例数据
x_data = np.linspace(-5, 5, 50)
y_data = 3 * np.sin(1.5 * x_data) + np.random.normal(size=50)

# 正弦波拟合函数
def sine_fit(x, A, B, C, D):
    return A * np.sin(B * x + C) + D

# 拟合数据
params, params_covariance = curve_fit(sine_fit, x_data, y_data, p0=[3, 1.5, 0, 0])

# 绘制原始数据和拟合曲线
plt.figure(figsize=(10, 6))
plt.scatter(x_data, y_data, label='Data')
plt.plot(x_data, sine_fit(x_data, *params), label='Fitted function')

plt.legend()
plt.show()

在这里插入图片描述

例2

import numpy as np
from scipy.optimize import curve_fit
import matplotlib.pyplot as plt

# 示例数据
x_data = np.linspace(0, 2*np.pi, 100)
y_data = 3 * np.sin(2 * x_data + 1) + 2

# 正弦波拟合函数
def sin_fit(x, A, B, C, D):
    return A * np.sin(B * x + C) + D

# 使用curve_fit进行拟合
params, params_covariance = curve_fit(sin_fit, x_data, y_data, p0=[2, 2, 0, 0])

# 绘制原始数据和拟合曲线
plt.figure(figsize=(10, 6))
plt.scatter(x_data, y_data, label='Data')
plt.plot(x_data, sin_fit(x_data, *params), label='Fitted function')

plt.legend()
plt.show()