用tushare和mplfinance画K线图

用tushare和mplfinance画K线图

导入 tushare并连接API获取股票数据

import tushare as ts
ts.set_token('Your token')
pro = ts.pro_api()
df= pro.daily(ts_code='600809.SH', start_date='20220401', end_date='20220701')

导入需要的包

import pandas as pd
import numpy as np
import mplfinance as mpf
import matplotlib.pyplot as plt
import talib
from matplotlib.dates import DateFormatter, WeekdayLocator, DayLocator, MONDAY,YEARLY
df1=df

 mplfinance对数据格式有要求需做一定的处理

def getData(df):df=pd.DataFrame(ame(columns={'trade_date':'Date','open':'Open','high':'High','low':'Low','close':'Close','vol':'Volume'},inplace=True)df=df[['Date','Open','High','Low','Close','Volume']]df['Date']&#_datetime(df['Date'])df.set_index(['Date'],inplace=True)df=df.sort_index()return df
df=getData(df)

 设置自己的绘画风格

#设置绘画风格
mystyle = dict(style_name    = 'style',base_mpl_style= 'fast', marketcolors  = {'candle'  : {'up':'w', 'down':'g'}, #填充颜色'edge'    : {'up':'r', 'down':'g'}, #边缘颜色'wick'    : {'up':'r', 'down':'g'}, #灯芯颜色'ohlc'    : {'up':'r', 'down':'g'},'volume'  : {'up':'r', 'down':'g'}, #交易量颜色'vcdopcod': True, 'alpha'   : 1.0,},mavcolors     = ['#ef5714','#ef5714','#9f4878','#9f4878'], #均线颜色y_on_right    = False,gridcolor     = '#a0a0a0',gridstyle     = '--',facecolor     = 'w',rc            = [ ('axes.edgecolor'  , 'white'   ),('axes.linewidth'  ,  1.5      ),('axes.labelsize'  , 'large'   ),('axes.labelweight', 'semibold'),(&#id'       ,  True     ),(&#id.axis'  ,  'y'      ),('grid.linewidth'  ,  0.4      ),('lines.linewidth' ,  2.0      ),('font.weight'     , 'medium'  ),('font.size'       ,  10.0     ),('figure.titlesize', 'x-large' ),('figure.titleweight','semibold'),],base_mpf_style= 'charles')
#加上字体的设置，否则中文会乱码
normal_font = {'fontname':'STZhongsong','size':     '12','color':    'green','va':       'bottom','ha':       'left'}

获得K线数据

#获得均线
def MA(df,k):close = [float(x) for x in df['Close']]x=talib.MA(np.array(close), timeperiod=k)return x
df1['MA3']=MA(df,3)
df1['MA10']=MA(df,10)
df1['MA21']=MA(df,21)

画图

#选择自己的绘画风格
fig = mpf.figure(figsize=(10, 6),style=mystyle) 
ax1 = fig.add_subplot(2, 1, 1)
ax2 = fig.add_subplot(2, 1, 2, sharex=ax1)
fig.subplots_adjust(hspace=0)   #子图之间没有间隔
plt.xlabel('date',fontsize=10)  #指定X轴标签和大小
plt.tick_params(labelsize=6)#指定X轴刻度字体大小
#为子图1添加均线
ax1.plot(df1['Date'],df1['MA3'],label='MA3',color='slategray')
ax1.plot(df1['Date'],df1['MA10'],label='MA10',color='skyblue')
ax1.plot(df1['Date'],df1['MA21'],label='MA21',color='#ef5714')
ax1.legend(fontsize=7)
#用mplfinance画出蜡烛图
mpf.plot(df, type='candle',ylabel='Candle', axtitle='my style',ax=ax1,volume=ax2)

结果如图：