matplotlib-在绘制时间序列中跳过无数据区间
主要参考:
https://blog.csdn.net/tpoy0099/article/details/49511907
https://blog.csdn.net/norsd/article/details/41020117
import numpy as np
import pandas as pd
from datetime import *
import matplotlib.pyplot as plt
import matplotlib.ticker asa mtk
# 一般性可用的方法:
# 输入 axis,
# x轴数据(type为DF or list)
# y轴数据(type为DF or list)
def decorateAx(ax, xs, ys):
def x_fmt_func(x, pos=None):
idx = np.clip(int(x + 0.5), 0, len(xs) - 1)
return xs[idx]
idx_pxy = np.arange(len(xs))
ax.plot(idx_pxy, ys, linewidth=1, linestyle="-")
ax.plot(ax.get_xlim(), [0, 0], color="blue", linewidth=0.5, linestyle="--")
ax.xaxis.set_major_formatter(mtk.FuncFormatter(x_fmt_func))
ax.grid(True)
return
df = pd.read_csv('./data.csv')
fig = plt.figure()
ax1 = fig.add_subplot(2,1,1)
ax2 = fig.add_subplot(2,1,2)
decorateAx(ax1, df.index, df['Pnl'].cumsum())
decorateAx(ax2, list(df.index)[-100:], df.loc[-100:, 'Pnl'].cumsum())