Python之获取每日涨停板股票数据并绘制K线图!

利用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.程序运行结果