获取数据
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')