写出用python做分位数回归以及分位数回归图的代码
时间: 2023-10-15 10:03:36 浏览: 117
以下是python实现分位数回归以及分位数回归图的代码:
```
import pandas as pd
import numpy as np
import statsmodels.formula.api as smf
import matplotlib.pyplot as plt
# 读取数据
data = pd.read_csv('data.csv')
# 分位数回归
model = smf.quantreg('y ~ x', data)
q = np.arange(0.05, 0.96, 0.1)
res = [model.fit(q=q[i]) for i in range(len(q))]
# 绘制分位数回归曲线
x = np.linspace(data.x.min(), data.x.max(), 100)
y_hat = np.array([res[i].predict({'x': x}) for i in range(len(q))])
plt.plot(x, y_hat.T, '-', lw=2, alpha=0.6)
# 绘制散点图
plt.scatter(data.x, data.y, alpha=0.5)
plt.xlabel('x')
plt.ylabel('y')
plt.title('Quantile Regression')
plt.show()
```
这段代码将根据给定数据集,绘制出分位数回归曲线和散点图。其中,利用statsmodels库中的quantreg函数进行分位数回归的拟合。其中,拟合结果保存在res列表中,然后利用matplotlib库绘制分位数回归曲线和散点图。
阅读全文