使用matplotlib绘制k线图(去掉空白日期)

之前研究过使用 matplotlib.finance 中的 candlestick_ohlc() 方法绘制k线图,但是有个很无语的问题,绘制出来的k线不是连续的——周末两天没有被去掉,因此总是留下一个空档。更不要说遇上春节这样的节假日,k线将留下很大一段无用的空白。非常影响观察k线走势。

K-BLANK.png

google查询了一阵,零零散散看到一些解决思路,某些代码可用;另外一些不知是因为当时使用的开源包和现在有所差异还是什么的,总之把大段代码复制过来测试,得到的只是一堆报错信息。不过,总归解决思路还是明确了。

解决思路

由于candlestick_ohlc() 方法内部是将一个连续的日期作为x轴的刻度送到matplotlib的绘图引擎中的,如果不是采用修改 candlestick_ohlc() 的源码,那么比较合理方法就是不要将日期数据送到 candlestick_ohlc() 方法中,并且重新自定义 x 轴的刻度。

股票数据重构

matplotlib 官方给出的candlestick_ohlc() 的推荐使用方式是这样:

mpf.candlestick_ochl(ax,data_mat,colordown='#53c156', colorup='#ff1717',width=0.3,alpha=1)

其中 ax 是绘制图形的 axis 对象,data_mat 是所有的股票数据。股票数据是一个二维矩阵,每一行都是按照 date,open,close,high,low,volume 的顺序排列的。这里 date 的值并不是 string,也不是 datetime,而是 pandas.TimeStamp。其实TimeStamp就是一个整型数字,类似于unix 系统中的 timestamp。

所以在构建股票数据时,date 这个位置我们可以将它赋值为从0开始的连续自然数,这样 candlestick_ochl() 方法绘图时,就不会把 date 转化为一个连续的日期(还包含周末那种)。所以,重构后的股票数据大致应该是这样:

[
  (0, 16.14, 16.24, 16.36, 16.14, 481999.28),
   (1, 16.24, 16.32, 16.38, 16.2, 424100.84), 
  (2, 16.32, 16.33, 16.39, 16.32, 276957.25), 
  (3, 16.3, 16.17, 16.38, 16.16, 277753.09)
]

每一行都是一个元组,元组里分别是 date, open, close, high, close 数据。

这样一来,绘制的图形就变成了:

K-BLANK2.png

可以看到,k线图形变得连贯了。但是x轴的刻度却变成了自然数,而非日期。所以,x 轴的刻度需要单独处理一下。

x轴刻度设定

假定所有的日期字符串都在 data['date'] 中,简单把所有日期数据甩给matplotlib,x轴的刻度就会密密麻麻的挤在一起。

ax.set_xticks(range(len(date_tickers)))
ax.set_xticklabels(date_tickers)
K-BLANK3.png

那么如何让 matplotlib 在绘图时只保留主要刻度呢?

如果只是这样:

ax.set_xticklabels(date_tickers)
K-BLANK4.png

乍一看,问题解决了!但是仔细一看,刻度不对!最后一个日期居然还是 2017-1-12 日,而k线已经是60天的数据了。

正确的姿势应该是用:

import matplotlib.ticker as ticker

# 先设定一个日期转换方法
def format_date(x,pos=None):
    # 由于前面股票数据在 date 这个位置传入的都是int
    # 因此 x=0,1,2,...
    # date_tickers 是所有日期的字符串形式列表
    if x<0 or x>len(date_tickers)-1:
        return ''
    return date_tickers[int(x)]

# 用 set_major_formatter() 方法来修改主刻度的文字格式化方式
ax.xaxis.set_major_formatter(ticker.FuncFormatter(format_date))

这样一来,就变成我想要的效果了:

K-BLANK5.png

但是,还有一点不满意,matplotlib自动生成的主刻度的间距,我认为太宽了。那么,我还可以用:

ax.xaxis.set_major_locator(ticker.MultipleLocator(6))

来强制指定每隔6个刻度,设定一个主刻度。图形效果就变成了这样:

K-BLANK6.png

完整的代码

import matplotlib.pyplot as plt
import matplotlib.finance as mpf
import numpy as np
import pandas as pd
from matplotlib.pylab import date2num
import matplotlib.ticker as ticker
import time

data=pd.read_csv(u'assets/兴业银行.csv',usecols=['date','open','close','high','low','volume'])
data[data['volume']==0]=np.nan
data=data.dropna()
data.sort_values(by='date',ascending=True,inplace=True)
# 原始的csv 读入进来 DataFrame 的 columns 顺序不符合candlestick_ochl 要求的顺序
# columns 的顺序一定是 date, open, close, high, low, volume
# 这样才符合 candlestick_ochl 绘图要求的数据结构
# 下面这个是改变列顺序最优雅的方法
data=data[['date','open','close','high','low','volume']]
data=data.head(62)

# 生成横轴的刻度名字
date_tickers=data.date.values

weekday_quotes=[tuple([i]+list(quote[1:])) for i,quote in enumerate(data.values)]
# print weekday_quotes

fig,ax=plt.subplots(figsize=(1200/72,480/72))

def format_date(x,pos=None):
    if x<0 or x>len(date_tickers)-1:
        return ''
    return date_tickers[int(x)]

ax.xaxis.set_major_locator(ticker.MultipleLocator(6))
ax.xaxis.set_major_formatter(ticker.FuncFormatter(format_date))
ax.grid(True)
# fig.autofmt_xdate()

mpf.candlestick_ochl(ax,weekday_quotes,colordown='#53c156', colorup='#ff1717',width=0.2)
plt.show()
最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 157,298评论 4 360
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 66,701评论 1 290
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 107,078评论 0 237
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 43,687评论 0 202
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 52,018评论 3 286
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 40,410评论 1 211
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 31,729评论 2 310
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 30,412评论 0 194
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 34,124评论 1 239
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 30,379评论 2 242
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 31,903评论 1 257
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 28,268评论 2 251
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 32,894评论 3 233
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 26,014评论 0 8
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 26,770评论 0 192
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 35,435评论 2 269
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 35,312评论 2 260

推荐阅读更多精彩内容