获取数据
import pandas as pd
import tushare as ts
import numpy as np
from sklearn.neighbors import KNeighborsClassifier
import matplotlib.pyplot as plt
import seaborn as sns
#首先我们来定义一个函数,用来获取数据
传入的三个参数分别是开始日期,结束日期和输出的文件名
def load_stock(start_date, end_date, output_file):
#首先让程序尝试读取已下载并保存的文件
try:
df = pd.read_pickle(output_file)
#如果文件已存在,则打印载入股票数据文件完毕
print('载入股票数据文件完毕')
#如果没有找到文件,则重新进行下载
except FileNotFoundError:
print('文件未找到,重新下载中')
#这里制定下载中国平安(601318)的交易数据
#下载源为yahoo
df = ts.get_k_data('601318', start_date, end_date)
df = df.set_index('date')
#下载成功后保存为pickle文件
df.to_pickle(output_file)
#并通知我们下载完成
print('下载完成')
#最后将下载的数据表进行返回
return df
#下面使用我们定义好的函数来获取交易数据
获取三年的数据,从2017年3月9日至2022年的12月25日
保存为名为601318的pickle文件
zgpa = load_stock(start_date = '2017-03-09',
end_date = '2022-12-25',
output_file = '601318.pkl')