利用tushare财经数据接口获取股市进行数据分析、并可视化的案例
1.导入所需要的库
import os
import pandas as pd
import tushare as ts
import time
import datetime
import xlrd
from dateutil.relativedelta import relativedelta #计算前多长时间间隔的日期的库
from matplotlib import pyplot as plt
import mpl_finance as mpf #画k线图的金融库
from matplotlib.pylab import date2num
2.获取当天日期和之前一年的日期
Python资源共享群：626017123
today = datetime.datetime.now().strftime('%Y%m%d')
t = datetime.datetime.strptime(today,'%Y%m%d').date()
year_before_today = (t-relativedelta(years=1)).strftime('%Y%m%d')
3.设置获取接口数据的token,在tushare官网注册用户获得
ts.set_token('752a393dbf58f03637ebc9c340c8e6ccb07322d3c5421ba680aae2eb')
4.设置每只股票全年数据文件路径
path = 'D://python_data_analysis//tushare_project'
if not os.path.exists('{0}//each_stock_year_data//{1}'.format(path, today)):
os.makedirs('{0}//each_stock_year_data//{1}'.format(path, today))
print("当日每只股票行情数据文件夹创建完成")
else:
print("当日每只股票行情数据文件目录存在")
5.设置每只股票K线图的存储路径
if not os.path.exists('{0}//k_line_graph//{1}'.format(path,today)):
os.makedirs('{0}//k_line_graph//{1}'.format(path,today))
print("当日每只股票的K线图存储路径文件夹创建完成")
else:
print("当日每日股票的K线图存储路径文件夹已存在")
6.#利用tushare获取每日股票行情数据
#定义一个获取股市每日数据写入Excel的函数
def get_stock_data_to_excel(day):
print("******开始获取股市每日行情数据******")
pro = ts.pro_api()
stock_daily_data = pro.daily(trade_date=day)
df = pd.DataFrame(stock_daily_data)
df.to_excel('{0}//daily_stock_data//data{1}.xlsx'.format(path,day))
print("******每日行情数据获取成功******")
7.定义一个获取每日股市数据中每日涨跌幅大于9%的股票（比率可以自由设置）
filedir = path + "//daily_stock_data//data20190118.xlsx"
def stock_filter_than_percent9(file):
print("******开始分析今日所有涨停板股票******")
data = xlrd.open_workbook(file)
table = data.sheets()[0] #获取Excel表的第几张
table_nrows = table.nrows
pct_chg = []
code = []
for row in range(table_nrows):
if row == 0:
continue
else:
if table.row_values(row)[9] > 9:
pct_chg.append(table.row_values(row)[9])
code.append(table.row_values(row)[1])
result_code_info = dict(zip(code,pct_chg))
print("今日涨幅大于9%的股票共有{}只".format(len(result_code_info)))
return result_code_info
8.根据获取的涨停板股票代码，获取其相应的一年行情数据，为后面k线做准备
def get_code_year_data():
print("开始获取今日所有涨停板股票时间段为一年的行情数据")
pro = ts.pro_api() #tushare数据接口
count = 1
for item in stock_code.keys():
print("******开始获取第{0}只股票行情数据******".format(count))
time.sleep(3) #每次请求前暂停
data = pro.daily(ts_code=item,start_date=year_before_today,end_date=today)
df = pd.DataFrame(data)
df.to_excel('{0}//each_stock_year_data//{1}//{2}.xlsx'.format(path,today,item))
count += 1
print("今日所有的涨停板股票年行情数据获取完成")
9.画每日涨停板股票一年的K线图
def draw_k_line():
print("开始画股票K线图")
datafiledir = os.listdir(path+'//each_stock_year_data//{0}//'.format(today))
for i in range(len(datafiledir)):
each_stock_code = datafiledir[i].split('.')[0] #截取相应股票代码
quotes = []
stock = pd.read_excel(path+'//each_stock_year_data//{0}//{1}'.format(today,datafiledir[i])) #利用pandas设置excel数据为DataFrame
stock_row = stock.shape[0] #获取数据表的行数
#获取每个交易日的开盘价，收盘价，最高价，最低价
for row in range(stock_row):
stock_trade_date = str(stock.loc[row,'trade_date'])
stock_trade_date_change = stock_trade_date[0:4] + '-' + stock_trade_date[4:6] + '-' + stock_trade_date[6:]
stock_trade_date_num = date2num(datetime.datetime.strptime(stock_trade_date_change,'%Y-%m-%d'))
stock_trade_date_plot = stock_trade_date_num #这个是画K线图必须使用的日期格式
stock_open_price = stock.loc[row,'open']
stock_close_price = stock.loc[row,'close']
stock_high_price = stock.loc[row,'high']
stock_low_price = stock.loc[row,'low' ]
#按照 candlestick_ohlc 函数要求的数据结构构造数据
datas = (stock_trade_date_plot,stock_open_price,stock_high_price,stock_low_price,stock_close_price)
quotes.append(datas)
#开始画K线图
fig,ax =plt.subplots(facecolor=(0,0.5,0.5),figsize=(12,8))
fig.subplots_adjust(bottom=0.1)
ax.xaxis_date()
plt.xticks(rotation=45)#日期显示的旋转角度
plt.title(each_stock_code)
plt.xlabel('time')
plt.ylabel('price')
mpf.candlestick_ohlc(ax,quotes,width=1.0,colorup='r',colordown='green')#上涨为红色，下降为绿色
plt.grid(True)
plt.savefig(path+'//k_line_graph//{0}//{1}.png'.format(today,each_stock_code),dpi=100,facecolor=(0,0.5,0.5))
plt.close(fig)
#plt.show()
print("所有股票K线图绘制完毕,请在图片路径进行查看")
10.定义程序入口
if __name__ == '__main__':
get_stock_data_to_excel(today) #调用函数获取当天行情数据
stock_code = stock_filter_than_percent9(filedir) #调用函数分析今日涨停板股票，返回股票代码
get_code_year_data() #根据返回的股票代码，获取其一年的行情数据
draw_k_line() #根据全年行情数据画对应股票的K线
11.程序运行结果
网友评论