matplotlib绘制K线图

来源:互联网 发布:linux vi保存退出 编辑:程序博客网 时间:2024/04/27 19:17

获得苹果的股票数据(2016/1/4-2017/4/20)如下表:

Date Open High Low Close 2016/1/4 102.610001 105.370003 102 105.349998 2016/1/5 105.75 105.849998 102.410004 102.709999 2016/1/6 100.559998 102.370003 99.870003 100.699997 2016/1/7 98.68 100.129997 96.43 96.449997 2016/1/8 98.550003 99.110001 96.760002 96.959999

绘制代码:

import matplotlib.pyplot as pltfrom matplotlib.dates import DateFormatter, WeekdayLocator,DayLocator, MONDAYimport pandas as pdimport matplotlib.dates as mdatesfrom matplotlib.finance import candlestick_ohlc#1.数据预处理#读取数据data=pd.read_csv("C:/Users/Administrator/Desktop/APPL/APPL.csv")#将时间数据转换为pandas的时间格式data["Date"]=pd.to_datetime(data["Date"])#将时间数据转换为matplotlib的时间格式data['Date'] = data['Date'].apply(lambda d: mdates.date2num(d.to_pydatetime()))#将数据框转换为元组tuples = [tuple(x) for x in data[['Date','Open','High','Low','Close']].values]#2.图像设置#获得fig和ax对象fig, ax = plt.subplots()#调节横坐标距离底部的长度fig.subplots_adjust(bottom=0.2)#设置横坐标标签的显示时间间隔(显示每周的周一)mondays = WeekdayLocator(MONDAY)ax.xaxis.set_major_locator(mondays)#设置横坐标刻度的显示间隔(每个刻度代表一天)alldays = DayLocator()ax.xaxis.set_minor_locator(alldays)#设置横坐标标签的显示样式,为“月、日”,Jan 12weekFormatter = DateFormatter('%b %d')ax.xaxis.set_major_formatter(weekFormatter)#显示每个刻度的标签#dayFormatter = DateFormatter('%d')# ax.xaxis.set_minor_formatter(dayFormatter)#3.绘图#烛台图绘制candlestick_ohlc(ax, tuples, width=0.6,colorup='r',colordown="g")#调整图像设置横坐标标签的显示样式,获得当前坐标轴plt.gca、获得x轴刻度标签get_xticklabels、转换角度rotation、旋转角度hori...entplt.setp(plt.gca().get_xticklabels(), rotation=45, horizontalalignment='right')plt.show()

结果图:
这里写图片描述

0 0
原创粉丝点击