作者:Lemon
来源:Python数据之道
Plotly中绘制三种经典的
股票交易图表(含视频讲解)
大家好,我是 Lemon 。
背景
前一段时间, Lemon 发了一期视频,分享了 Plotly 和 Dash 在投资领域的应用案例。
点击上面图片查看视频
看完视频后,有同学提了一个问题:
股票价格曲线,带可调节的时间条的图怎么绘制?
今天 Lemon 来详细的分享下,这类图如何绘制,一共会讲解 3 类图形,分别是 面积曲线图、蜡烛图、OHLC图。这三种类型的图在投资中会经常遇到。
Lemon 录制了一个视频,来说明通过本文绘制的图表效果:
数据来源
本文的数据来自开源项目 tushare, 从 tushare 中获取数据,首先要进行注册获取 token(一串字母和数字组成的文本),然后才可以获取到数据,大家可以通过以下链接来注册(也可以点击文末的“阅读原文”):
https://tushare.pro/register?reg=129033
数据获取
在 tushare
中注册后,通过 “个人主页”——“接口TOKEN” 可以找到自己的 token 值,界面如下:
复制 token 值,然后在代码中进行如下设置:
# 设置 token
# tushare 注册地址:https://tushare.pro/register?reg=129033
# 以上方法只需要在第一次或者token失效后调用,完成调取tushare数据凭证的设置,正常情况下不需要重复设置。
ts.set_token('你的token值')
pro = ts.pro_api()
在设置好 token 值后,我们就可以开始获取数据,这里以获取沪深300指数为例,来演示三种类型的图形绘制。
数据获取及初步整理如下:
# plotly-fin-candlestick.ipynb
# 获取沪深300指数行情数据, asset='I', I 表示 index,即指数
# d_today = dt.date.today().strftime('%Y%m%d')
df_300 = ts.pro_bar(
ts_code='000300.SH',
asset='I',
start_date='20180101',
# end_date=d_today,
end_date='20210119'
)
# 设置时间格式
df_300['trade_date'] = pd.to_datetime(df_300['trade_date'])
df_300 = df_300.set_index('trade_date')
df_300
面积曲线图
在常见的股票软件时,查看分时图,大部分是使用的面积曲线图,这类图形是展示股票数据常用的类型之一。
默认的面积曲线图
在 Plotly 中,可以使用 plotly express 的 area
图来绘制面积曲线图。
代码如下:
# 面积曲线图
area_chart = px.area(df_300['close'], title = '沪深300')
area_chart.update_xaxes(title_text = '日期')
area_chart.update_yaxes(
title_text = '沪深300收盘价',
# tickprefix = '$'
)
area_chart.update_layout(showlegend = False)
# Pycharm,VS Code, Spider 等模式下
# Jupyter Notebook 下也可以用
# plot(area_chart,filename='tmp/hushen300-area.html')
# 在 J