用python绘制一元二次和一元三次模型的曲线图
时间: 2024-09-22 18:08:55 浏览: 22
在Python中,我们可以使用`matplotlib`库来绘制一元二次和一元三次模型的曲线图。以下是一个简单的例子,假设我们已经有了之前生成的一元二次和一元三次回归模型的参数:
```python
import numpy as np
import matplotlib.pyplot as plt
# 假设我们有以下模型参数:
quad_coeffs = [1, 2, 3] # 对应形如 y = ax^2 + bx + c 的一元二次模型
cub_coeffs = [4, 5, 6, 7] # 对应形如 y = ax^3 + bx^2 + cx + d 的一元三次模型
# 定义x的范围
x_values = np.linspace(start=-10, stop=10, num=400)
# 计算对应的y值
y_quad = quad_coeffs[0] * x_values**2 + quad_coeffs[1] * x_values + quad_coeffs[2]
y_cub = cub_coeffs[0] * x_values**3 + cub_coeffs[1] * x_values**2 + cub_coeffs[2] * x_values + cub_coeffs[3]
# 绘制图表
plt.figure(figsize=(8, 6))
plt.plot(x_values, y_quad, label="一元二次模型", color='blue')
plt.plot(x_values, y_cub, label="一元三次模型", color='red')
# 添加标题和标签
plt.title("一元二次和一元三次函数示意图")
plt.xlabel('X')
plt.ylabel('Y')
# 显示图例
plt.legend()
# 显示图形
plt.grid(True)
plt.show()
```
这个代码会创建一个图表,显示一元二次函数(y = ax^2 + bx + c)和一元三次函数(y = ax^3 + bx^2 + cx + d)随着x变化的情况,并且给每个函数添加了图例。