python调用tushare库做简单回归分析

1. 调用接口

tushare文档

2.获取数据并保存到excel

import tushare as ts
import pandas as pd

# ts.set_token("这里写你的token") 
ts.set_token("81665c80335ebe3ce0c417591062065be7f968867429309ae350")

pro=ts.pro_api()

df_000001 = pro.daily(ts_code='000001.SZ')
df_000002  = pro.daily(ts_code='000002.SZ')

# print(df_000001.head())

writer=pd.ExcelWriter(r"D:\Users\ZYZ\Desktop\金融数据.xlsx")

df_000002.to_excel(writer,"Sheet2")
df_000001.to_excel(writer,"Sheet1")

writer.save()

3.Skleran库线性回归

import matplotlib.pyplot as plt
import pandas as pd
from sklearn.linear_model import LinearRegression 


Ready_Data_Frame=pd.read_excel(r"D:\Users\ZYZ\Desktop\金融数据.xlsx","Sheet1")

# print(Ready_Data_Frame)

X=Ready_Data_Frame.iloc[:,6].values.reshape(-1,1)

Y=Ready_Data_Frame.iloc[:,10].values.reshape(-1,1)

LR_Model=LinearRegression() # 初始化模型

LR_Model.fit(X,Y)
 
Y_Predict=LR_Model.predict(X)


plt.scatter(X,Y)
plt.plot(X,Y_Predict,color="r")

plt.show()

4.结果展示