1 ##自选股（代码、日期）合并保存 2 import tushare as ts 3 import pandas as pd 4 ##输入自选股 5 data_list=['600850','603722','002411','002444','002456','300011'] 6 StockPrices = pd.DataFrame() 7 for num_stock in data_list: 8 rescore=ts.get_hist_data(num_stock,start='2020-01-05',end='2020-02-26',ktype='D')##选择日期 9 rescore['stock_name']=num_stock 10 rescore=rescore.reset_index() 11 StockPrices=StockPrices.append(rescore) 12 StockPrices.to_excel(excel_writer='D:/stock_ana/all.xlsx',index=False) 13 Sp=pd.DataFrame(StockPrices,columns=(['stock_name','date','close'])) 14 pivoted=Sp.pivot('date','stock_name','close') 15 import matplotlib.pyplot as plt 16 fig=plt.figure() 17 plt.figure(figsize=(30,10)) 18 l=len(data_list) 19 color_list=['r','g','b','y','k','tan','c','olive','pink','purple'] 20 for i,m in zip(data_list,range(l)): 21 plt.plot(pivoted[i],'ko--',color=color_list[m]) 22 plt.tick_params(labelsize=13)##字体大小 23 plt.xticks(rotation=45)##旋转角度 24 plt.legend(loc='best')##标签位置 25 plt.savefig(r'D:/stock_ana/自选股分析.jpg')
