最小二乘法拟合多项式的python代码

以下是使用最小二乘法拟合多项式的 Python 代码：

import numpy as np
from numpy.linalg import inv

# 生成随机数据
x = np.array([1, 2, 3, 4, 5])
y = np.array([0.5, 2.5, 2.0, 4.0, 3.5])

# 多项式拟合
n = 3 # 多项式次数
X = np.array([x**i for i in range(n+1)]).T
beta = np.dot(np.dot(inv(np.dot(X.T, X)), X.T), y)

# 预测
x_pred = np.linspace(0, 6, 50)
X_pred = np.array([x_pred**i for i in range(n+1)]).T
y_pred = np.dot(X_pred, beta)

# 绘图
import matplotlib.pyplot as plt
plt.plot(x, y, 'o', label='data')
plt.plot(x_pred, y_pred, label='prediction')
plt.legend()
plt.show()

这里我们拟合了一个三次多项式，并用生成的模型预测了新的数据点。可以根据需要修改 `n` 的值以及预测的数据范围。

相关问题

最小二乘法多项式拟合python代码

以下是使用最小二乘法进行多项式拟合的Python代码示例：

import numpy as np
import matplotlib.pyplot as plt

# Generate some sample data
x = np.array([1, 2, 3, 4, 5])
y = np.array([2.1, 3.8, 3.9, 5.2, 6.4])

# Define the degree of the polynomial to fit
degree = 2

# Perform the polynomial fit using numpy's polyfit function
coefficients = np.polyfit(x, y, degree)

# Create a function using the fitted coefficients
poly_func = np.poly1d(coefficients)

# Create a plot of the original data and the fitted function
plt.scatter(x, y, color='blue')
plt.plot(x, poly_func(x), color='red')
plt.show()

在上面的代码中，我们首先生成了一些样本数据，然后指定了要拟合的多项式的次数。然后，我们使用numpy的polyfit函数执行最小二乘拟合，并使用其返回的系数创建一个函数。最后，我们绘制了原始数据和拟合函数的图形。

python最小二乘法拟合多项式

Python中可以使用NumPy库中的polyfit函数进行最小二乘法拟合多项式。该函数的语法为：numpy.polyfit(x, y, deg, w)，其中x和y为拟合数据的自变量和因变量，deg为拟合多项式的阶次，w为拟合数据的权重（可选参数，默认为None）。该函数返回一个包含拟合多项式系数的一维数组。例如，要拟合一个二次多项式，可以使用以下代码：

import numpy as np

# 定义拟合数据
x = np.array([1, 2, 3, 4, 5])
y = np.array([2.1, 3.9, 6.1, 8.2, 10.1])

# 进行二次多项式拟合
coefficients = np.polyfit(x, y, 2)

# 输出拟合多项式系数
print(coefficients)

输出结果为：[ 0.16071429 -0.44642857 2.78571429]，表示拟合多项式为f(x) = 0.1607x^2 - 0.4464x + 2.7857。