- VS17安装Python的第三方包总是出错,改用软件管家下载到的VS2019,安装很方便,这个公众号真心不错
- 做了一个CSDN现有例子的测试,代码如下:
import pandas as pd
import numpy as np
import math
import matplotlib.pyplot as plt
from sklearn.linear_model import LinearRegression
from sklearn.model_selection import train_test_split
origDf = pd.read_csv('C:/StockExcel/603601.csv',encoding='gbk')
df = origDf[['Close', 'High', 'Low','Open' ,'Volume']]
featureData = df[['Open', 'High','Low', 'Volume']]
feature = featureData.values
target = np.array(df['Close'])
feature_train, feature_test, target_train ,target_test = train_test_split(feature,target,test_size=0.05)
pridectedDays = int(math.ceil(0.05 * len(origDf)))
lrTool = LinearRegression()
lrTool.fit(feature_train,target_train)
predictByTest = lrTool.predict(feature_test)
index=0
while index < len(origDf) - pridectedDays:
df.loc[index,'predictedVal']=origDf.loc[index,'Close']
df.loc[index,'Date']=origDf.loc[index,'Date']
index = index+1
predictedCnt=0
while predictedCnt<pridectedDays:
df.loc[index,'predictedVal']=predictByTest[predictedCnt]
df.loc[index,'Date']=origDf.loc[index,'Date']
predictedCnt=predictedCnt+1
index=index+1
plt.figure()
df['predictedVal'].plot(color="red",label='predicted Data')
df['Close'].plot(color="blue",label='Real Data')
plt.legend(loc='best')
major_index=df.index[df.index%10==0]
major_xtics=df['Date'][df.index%10==0]
plt.xticks(major_index,major_xtics)
plt.setp(plt.gca().get_xticklabels(), rotation=30)
plt.grid(linestyle='-.')
plt.show()
- 这段代码调试时,按照编译提示,把import中的第三方库都可以在python环境中搜索,可以自动安装

- 调试的问题是,原作者使用了df.ix,一直出现该属性不存在,后改为df.loc,据说是最新版本的pandas已经取消了ix方法
- 最后运行结果
