python+tushare+mplfinance绘制一只股票的K线图

最新推荐文章于 2024-01-21 21:28:20 发布

原创最新推荐文章于 2024-01-21 21:28:20 发布 · 1.8k 阅读

20 ·

CC 4.0 BY-SA版权

文章标签：

#python #数据可视化

量化选股专栏收录该内容

10 篇文章

订阅专栏

本文介绍如何使用Python的tushare、pandas和mplfinance库获取股票数据，并绘制指定时间段内的K线图和成交量图。代码实现了从网络获取股票代码为000001.SZ的股票数据，包括开盘价、收盘价、最高价、最低价和成交量，并将其转化为易于绘图的数据格式。

获取一个股票的某段时间内的开盘价、最高价、最低价、收盘价、成交量绘制出K线图及成交量图。

import tushare as ts
import pandas as pd
import matplotlib.pyplot as plt
import mplfinance as mpf
from datetime import datetime
 
ts.set_token('you token')
pro = ts.pro_api()
df = pro.daily(ts_code='000001.SZ', start_date='20200401', \
end_date='20201210')
df['date'] = df['trade_date'].apply\
(lambda x: datetime.strptime(x, '%Y%m%d'))   #将字符串型日期转换成时间戳
data = df.loc[:,['date', 'open','close','high','low','vol']]
data = data.rename(columns={"vol": "volume"})   # 修改列名
data.set_index('date', inplace=True)
data.index.name = "Date"
data = data.sort_index()

mc = mpf.make_marketcolors(up='r',down='g')
s  = mpf.make_mpf_style(marketcolors=mc,\
mavcolors=['#4f8a8b','#fbd46d','#87556f'])
 mpf.plot(data,type='candle',volume=True, \
show_nontrading=True, figratio=(20,10), mav=(5,10,20),style=s)