使用matplotlib绘制不连续时间序列的OHLC数据

引言：当我们需要绘制时间序列不连续的OHLC(Open,High,Low,Close)折线图时，使用pandas的plot函数就会将两个时间链接在一起，形成一段长长的直线。 So Ugly! Follow this：



如何解决这个问题呢？

加载数据

定义映射函数

plot轻松绘图

AU1806下载

加载数据

从文件AU1806中读取二进制数据，转为pandas.DataFrame

import pandas as pd
import matplotlib.pyplot as plt
import numpy as np

pd.set_option('display.expand_frame_repr',False)
df=pd.read_pickle('AU1806')
rng_date = df.index
df = df.reset_index()

数据一览

定义映射函数

我们按照：’year’,’month’,’day’,’hour’,’minute’,’second’ 频率生成等间隔 frequency 时间列，并以此分组排序，取每一组的最后一项的index值和frequency值，保存在列表。

from collections import OrderedDict

def tick_category(frequency, step):
'''
根据频率和频数获取刻度的时间轴
frequency 可取值: 'year','month','day','hour','minute','second'
'''

if frequency=='year':
df['frequency'] = rng_date.strftime('%Y')
elif frequency=='month':
df['frequency'] = rng_date.strftime('%Y-%m')
elif frequency=='day':
df['frequency'] = rng_date.strftime('%Y-%m-%d')
elif frequency=='hour':
df['frequency'] = rng_date.strftime('%Y-%m-%d %H')
elif frequency=='minute':
df['frequency'] = rng_date.strftime('%Y-%m-%d %H:%M')
elif frequency=='second':
df['frequency'] = rng_date.strftime('%Y-%m-%d %H:%M:%S')
else:
df['frequency'] = rng_date.strftime('%Y')

num_date = OrderedDict()
for item in df.groupby('frequency'):
num_date[item[1].index[-1]] = item[0]

nums = list(num_date.keys())[::step]
dates = list(num_date.values())[::step]
return num_date,nums,dates

plot轻松绘图

为了便于细致的控制界面，直接采用matplotlib绘图。

def show_plot(frequency, step, columns, tick_count=10):
num_date,nums,dates = tick_category(frequency, step)
end_pos = nums[tick_count] if tick_count<len(nums) else nums[-1]
data = df.loc[:,columns][:end_pos]
axes = data.plot()
axes.set_xticks(nums[:end_pos])
axes.set_xticklabels(dates[:end_pos],rotation=45)
plt.show()

效果预览：

show_plot('year',1, ['open', 'high','low','close'], tick_count=10)

show_plot('month',1, ['open'], tick_count=10)

你看懂是怎么计算的了吗？是不是很简单

AU1806下载

AU1806下载地址