上一篇《用Python绘制专业的K线图》,讲解了数据获取、K线图绘制及成交量绘制等内容。本篇将在上一篇的基础上,继续讲解移动均线的绘制。
1、获取数据
我们从恒有数金融数据社区,获取股票市场历史行情数据。我们获取2021年3月1号至2021年6月1号,恒生电子(600570.SH)的日行情数据,并做简单处理,代码及执行结果如下。
# 加载取数与绘图所需的函数包
import pandas as pd
import datetime
from hs_udata import set_token,stock_quote_daily
from mpl_finance import candlestick_ohlc
import matplotlib as mpl
import matplotlib.pyplot as plt
import matplotlib.dates as mdates
mpl.rcParams['font.sans-serif'] = ['SimHei'] # 指定默认字体
mpl.rcParams['axes.unicode_minus'] = False # 解决保存图像是负号'-'显示为方块的问题
def GetData(stock_code,start,end):
#stock_code:获取股票数据的股票代码
# start:开始日期
# end:结束日期
date_start=datetime.datetime.strptime(start,'%Y-%m-%d')
date_end =datetime.datetime.strptime(end,'%Y-%m-%d')
data = pd.DataFrame([])
while date_start<date_end:
# 获取日行情数据,接口说明见 https://udata.hs.net/datas/332/
# adjust_way枚举值为:0-不复权,1-前复权,2-后复权,此处取前复权
data_i = stock_quote_daily(en_prod_code=stock_code
,trading_date=date_start.strftime('%Y%m%d')
,adjust_way = 1)
data=pd.concat([data,data_i],axis=0) # 将行情数据按行拼接
date_start+=datetime.timedelta(days=1) # 日期变量自增
# 返回行情数据
return data
#1、获取行情数据
stock_code =