python用matplotlib画图例子(2)

背景：

代码

import numpy as np
import matplotlib.pyplot as plt
import pandas as pd
from datetime import datetime
"""
数据格式（2017沪深300.CSV）：
date    HS300
2017/1/3    3342.2272
2017/1/4    3368.3117
2017/1/5    3367.7892
………………
"""
filepath = 'C:\\Users\\tinysoft\\Desktop\\2017沪深300.CSV'
data = pd.read_csv(filepath,index_col=0,parse_dates = True)

hs300 = data['HS300']

fig = plt.figure()
ax = fig.add_subplot(111)

hs300.plot(ax=ax,style='r')

"""
设置标注：
日期
备注文本
颜色
"""
day_label = [
(datetime(2017,10,1),'National Day','r'),#国庆节
(hs300.idxmin(),"min",'g') #2017年最低
]

for day,text,color in day_label:
label = "{0}\n{1}\n{2}".format(day.date(),hs300.asof(day),text)
ax.annotate(label,
xy = (day,hs300.asof(day)+10),#数据位置
xytext = (day,hs300.asof(day)+150),#字体位置
arrowprops = dict(facecolor=color),#箭头颜色
horizontalalignment='left',#水平对齐方式
verticalalignment='top')#垂直对齐方式

ax.set_xlim([datetime(2017,1,1),datetime(2017,12,31)])

ax.set_ylabel('$Close$')
ax.set_xlabel('$SH000300$')
ax.set_title('$In\ 2017, the\ SH000300\ charts$')

plt.show()

结果