引入类库
import matplotlib as mplimport tushare as tsimport matplotlib.pyplot as pltimport matplotlib.finance as mpffrom matplotlib import rcfrom matplotlib.pylab import date2numfrom datetime import datetime, timedelta
设置取k线的时间
today = datetime.now().strftime('%Y-%m-%d')before = (datetime.now() - timedelta(days = 120)).strftime('%Y-%m-%d') #120自然日，大概80个交易日
获取行情数据
#获取行情数据，格式： pandas.core.frame.DataFramecode = 'tsla_df = ts.get_k_data(code, start = before, end = today)# print(tsla_df)
'''蜡烛图的日期，不支持普通的YYYY-MM-DD格式要使用matplotlib.finance.date2num进行转换为特有的数字值'''qutotes = []for _, (d, o, c, h, l) in enumerate( zip(tsla_df.date, tsla_df.open, tsla_df.close, tsla_df.high, tsla_df.low)): d = mpf.date2num(datetime.strptime(d,'%Y-%m-%d')) # 日期，开盘，收盘，最高，最低组成tuple对象val val = (d, o, h, l, c) # 加val加入qutotes qutotes.append(val)
fig, ax = plt.subplots(figsize=(15,5)) #设置图片大小fig.subplots_adjust(bottom=0.5) #调整画框的位置，用来消除白边ax.xaxis_date() # X轴的刻度为日期plt.xticks(rotation=45) # 设置日期刻度旋转的角度plt.title(code) # 设置图片标题plt.xlabel('Date') # 设置X轴标题plt.ylabel('Price') # 设置Y轴标题plt.grid(True) # 显示网格mpf.candlestick_ohlc(ax, qutotes, width=0.6, colorup='g', colordown='r', alpha=1.0)plt.show()
