matplotlib 股票走势图 瑞幸咖啡

 

本文主要用折线图描绘瑞幸咖啡19年上市以来的股价走势，再利用线条颜色、标注、箭头等突出财务造假丑闻对估价的影响。

数据源：雅虎财经信息。如何获取股价信息可以看这里：https://www.cnblogs.com/chenboshi/p/13539918.html

                              

画图思路：

1.选取收盘价和时间这两个字段，x轴为交易日，y轴为收盘价；

下载的数据集日期只有股票交易日，绘制时间轴时，不填充节假日，直接进行按照月份聚合。

2.以4月1日（财务造假丑闻爆出）作为界限，用不同颜色标记。4月1日前用深蓝色画图，因为瑞幸咖啡品牌颜色是深蓝色。4月1日后用红色。

3.添加虚横线，利用刻度对比估价涨跌幅度；

4.添加文本标注和箭头，指出导致股价暴跌的事件和时间点；

5.添加长方形阴影，再加个4月1日前后的对比。完事ヽ(￣▽￣)و

 

 

 

代码地址  →→  github  

 

import matplotlib.patches as patches
import matplotlib.pyplot as plt
import pandas as pd
import numpy as np
plt.rcParams['font.sans-serif'] = ['SimHei'] # 指定默认字体
plt.rcParams['axes.unicode_minus'] = False # 解决保存图像是负号'-'显示为方块的问题

#导入数据
lk= pd.read_csv("C:\mypython\datasets\stock-LK.csv",encoding='utf-8')

# 自上市以来 收盘价最低、最高点
print(np.min(lk['Close']))
print(np.max(lk['Close']))
# 将时间转换成 yyyy-mm格式
lk['Ymnth']=lk['Date'].apply(lambda x : x[:7])

# 以2020-04-01为界限区分财务造假新闻爆出前后的股价走势
before=lk[lk['Date']<'2020-04-01']
after=lk[lk['Date']>='2020-04-01']

fig, ax = plt.subplots(1,1,figsize=(16,9), dpi= 300)
ax.plot(before['Date'], before['Close'], color='#142975',lw=2)
ax.plot(after['Date'], after['Close'], color='#B00E11',lw=2)

#y轴刻度和标签设置
ax.set_ylim(0,55)
ax.set_yticklabels(labels=range(0,61,10),fontsize=14,fontweight='medium')
for y in range(0,55,10):
    plt.axhline(y=y,color='gray',linestyle='--',alpha=0.3)
    
#日期缺失非工作日和节假日，可以选择填充，或者忽略x轴上每月总天数长度，进行聚合。此处选择后者
#首先按照年月分组，选取每月最小值即第一个股票交易日 作为刻度 比如
xticks=lk[['Date','Ymnth']].groupby(lk['Ymnth']).agg('min')['Date'].tolist()  
ax.set_xticks(xticks)
#print(xticks)  #['2019-05-17', '2019-06-03', '2019-07-01'...]
xtick_color= [ '#142975' if x <'2020-04-01' else '#B00E11' for x in xticks ]

#然后刻度标签设置为 上述列表中每个刻度的前四位  ['2014', '2015', '2016',...
ax.set_xticklabels([x[:7] for x in xticks ], rotation=30,fontsize=14)
# 根据时间分割线，设置不同的刻度标签颜色
[t.set_color(i) for (i,t) in zip(xtick_color,ax.xaxis.get_ticklabels())]

# 文本标注
box = {
  'facecolor'  : 'white',
  'edgecolor' : 'red',
  'boxstyle'    : 'round'
}
ax.text('2020-04-15',25,"爆出财务造假丑闻",bbox=box,color='red',fontsize=20)

# 箭头指向文本  文本此物为空格. 4月1日估价为26块多
plt.annotate(' ',
ha = 'center', va = 'bottom',
xytext = ('2020-04-07', 26),
xy = ('2020-04-01', 26),
arrowprops = {'color':'red'})

#给红色部分加上灰色阴影
start=before.shape[0]
offset=after.shape[0]+50
shape = patches.Rectangle((start,-5),offset,60,color = '#E9ECEF')
plt.gca().add_patch(shape)

plt.title("瑞幸咖啡历史收盘价",fontsize=26)

fig.tight_layout()
plt.savefig("C:\mypython\plots\stock_lk4.png")

plt.show()
plt.close()