捕捉趋势最普遍的方法为移动平均线,根据求平均的方式不同,移动平均数又可分为简单移动平均数(Simple Moving Average, SMA),加权移动平均数(Weighted Moving Average, WMA),和指数移动平均数(Exponential Moving Average, EXPMA或EMA)。
以从Tushare获取深证成指(‘399001.SZ’)近100交易日数据为例。
# 导入相关模块
import os
import numpy as np
import tushare as ts
import pandas as pd
import matplotlib.pyplot as plt
from datetime import datetime, timedelta
# from pylab import mpl
token = 'Your token' # 输入你的接口密匙,获取方式及相关权限见Tushare官网。
pro = ts.pro_api(token)
df = pro.index_daily(ts_code='399001.SZ') #index_daily为指数日线的接口。股票的则为daily。
# 将获取到的DataFrame数据进行标准化处理,转换为方便自己使用的一种规范格式。
df = df.loc[:, ['trade_date', 'open', 'high', 'low', 'close', 'vol']]
df.rename(
columns={
'trade_date': 'Date', 'open': 'Open',
'high': 'High', 'low': 'Low',
'close': 'Close', 'vol': 'Volume'},
inplace=True) # 重定义列名,方便统一规范操作。
df['Date'] = pd.to_datetime(df['Date']) # 转换日期列的格式,便于作图
df.set_index(['Date'], inplace=True) # 将日期列作为行索引
df = df.sort_index() # 倒序,因为Tushare的数据是最近的交易日数据显示在DataFrame上方,倒序后方能保证作图时X轴从左到右时间序列递增。
df1 = df[-100:] # 取出近100交易日的数据。
Close = df1['Close'] # 取出收盘数据。
now = datetime.now() # 现在时间
today = now.strftime('%Y-%m-%d') # 今日日期
# 设置一会儿生成的图像文件的保存路径,这里该三行命令在D盘创建一个以今日日期命名的文件夹。
path = 'D:/' + today
if not os.path.exists(path):
os.mkdir(path)
# 准备工作完毕,走起!
# 求取5日简单移动平均数
Sma5 = pd.Series(0.0,index=Close.index)
for i in range(4,len(Close)):
Sma5[i] = sum(Close[i-4:(i+1)])/5 # Sma5的前四个数据为0,下边使用时需要切掉。
# 求取10日简单移动平均数
Sma10 = pd.Series(0.0,index=Close.index)
for i in range(9,len(Close)):
Sma10[i] = sum(Close[i-9:(i+1)])/10 # Sma10的前9个数据为0,下边使用时需要切掉。
# 绘图 (绘制出的图像中,因为均线的起点有滞后,所以均线数据并不足上边\
说的100交易日,但也因此可以根据起点先后快速判断出均线的时间跨度长短)
plt.rcParams['font.sans-serif'] = ['SimHei']
plt.plot(Close, label="Close", color='g')
plt.plot(Sma5[4:], label="Sma5", color='r', linestyle='dashed')
plt.plot(Sma10[9:], label="Sma10", color='b', linestyle='dashed')
plt.title('深证成指近100交易日收盘数据附5&10日MAV图')
plt.legend() # 增设图例
plt.show()
# plt.savefig(path + '/深证成指收盘数据时序图附5&10日MAV.png')
def smaCal(tsPrice, k):
import pandas as pd
Sma = pd.Series(0.0, index=tsPrice.index)
for i in range(k-1, len(tsPrice)):
Sma[i] = sum(tsPrice[(i-k+1):(i+1)])/k
return(Sma)
# 调用
smaCal(Close, 5)
# 设定传入三个参数:name:序列名字(如“深证成指”,便于注明于图像标题。)
# tsPrice: Series类型的数据序列。
# 不定参数规定为最多五个才能绘图。
def plot_smaCal(name,tsPrice, *args):
if len(args)>5:
print('提示:添加均线数量最多是五个!')
return None
import pandas as pd
plt.plot(tsPrice, label="Close", color='g')
colors = ['r', 'k', 'b', 'c', 'y'] # 设定5个颜色,也即关系到上边均线数量的上限。
j = 0 # 用于颜色索引转换
for k in args:
Sma=pd.Series(0.0, index=tsPrice.index)
for i in range(k-1, len(tsPrice)):
Sma[i] = sum(tsPrice[(i-k+1):(i+1)])/k
plt.plot(Sma[k-1:], label="Sma"+str(k), color=colors[j], linestyle='dashed')
j += 1
days='&'.join([str(ints) for ints in args])
plt.title(name + '收盘数据附' + days + '日MAV图')
plt.legend() # 增设图例
plt.show()
# plt.savefig(path + '/' + name + '收盘数据时序图附' + days + 'MAV.png')
# 调用
plot_smaCal('深证成指',Close,5,10,20)
直接使用mplfinance库来绘制不失为一种更简单的方法。
import mplfinance as mpf
road = path + '/深证成指' + '深证成指近100交易日收盘数据附5&10日MAV图'
mpf.plot(df1, type='line',
ylabel="price", style='blueskies', mav=(5, 10),
volume=True, ylabel_lower="volume(shares)", savefig=road)
# 通过设定不定长参数mav,即想要得到简单移动平均线的时间跨度,来获取想要的结果。
然后在前边指定的目录即可查看到图像
加权移动平均,即对数据赋予一定的权重后再求平均。
一般认为离当前时间点越近的数据越具有代表性,越远则越没有代表性。
WMAt=5 = w1p1 + w2p2 + w3p3 + w4p4 + w5p5
(其中w1 + w2 + w3 + w4 + w5 = 1)
# 5日加权移动平均线
b1 = np.array([1, 2, 3, 4, 5])
w1 = b1/sum(b1) # 定义一组权重
Wma5 = pd.Series(0.0, index=Close.index)
for i in range(4,len(Close)):
Wma5[i] = sum(w1*Close[i-4:(i+1)])
# 10日加权移动平均线
b2 = np.array([1, 2, 3, 4, 5, 6, 7, 8, 9, 10])
w2 = b2/sum(b2) # 定义一组权重
Wma10 = pd.Series(0.0, index=Close.index)
for i in range(9,len(Close)):
Wma10[i] = sum(w2*Close[i-9:(i+1)])
def wmaCal(tsPrice, weight):
import pandas as pd
import numpy as np
k = len(weight)
arrWeight = np.array(weight)
Wma=pd.Series(0.0, index=tsPrice.index)
for i in range(k-1, len(tsPrice.index)):
Wma[i] = sum(arrWeight * tsPrice[(i-k+1):(i+1)])
return(Wma)
# 调用
b = np.array([1, 2, 3, 4, 5])
w = b1/sum(b)
wmaCal(Close,w)
plt.rcParams['font.sans-serif'] = ['SimHei']
plt.plot(Close, label="Close", color='g')
plt.plot(Wma5[4:], label="Wma5", color='r', linestyle='dashed')
plt.plot(Wma10[9:], label="Wma10", color='b', linestyle='dashed')
plt.title('深证成指近100交易日收盘数据附5&10日加权WAV图')
plt.legend() # 增设图例
plt.show()
# plt.savefig(path + '/深证成指收盘数据时序图附5&10日MAV.png')
指数加权移动平均也可简称指数移动平均,相当于一种比较特别的加权移动平均。需要先给定一个权重值w,比如0.2。时间跨度为k。第k期的EWMAt=k为前k期的算数平均数。
第k+1期EWMAt=k+1 = w * pk+1 + (1-w) * EWMA t=k
即当天指数移动平均数 = 0.2*当天股价 + 0.8 * 前一天指数移动平均
后边的按此公式以此类推。
# 第五天指数加权移动平均数为前五天的算数平均数
Ewma5_number1 = np.mean(Close[0:5])
Ewma5 = pd.Series(0.0, index=Close.index)
Ewma5[4]=Ewma5_number1
# 计算第六天及以后的指数移动平均数
for i in range(5, len(Close)):
Ewma5[i] = Ewma5[i-1]*0.8 + Close[i]*0.2
Ewma5.head(10)
# 绘制指数加权移动平均线图像
plt.rcParams['font.sans-serif'] = ['SimHei']
plt.plot(Close, label="Close", color='r')
plt.plot(Ewma5[4:], label="Ewma5", color='g', linestyle='-.')
plt.title('深证成指近100交易日收盘数据附5日指数移动平均线')
plt.legend()
plt.show()
def ewmaCal(tsPrice, period=5,exponential=0.2):
import pandas as pd
import numpy as np
Ewma = pd.Series(0.0, index=tsPrice.index)
Ewma[period-1]=np.mean(tsPrice[:period])
for i in range(period, len(tsPrice)):
Ewma[i] = exponential * tsPrice[i] + (1 - exponential) * Ewma[i-1]
return(Ewma)
# 调用
Ewma=ewmaCal(Close,5,0.2)
Ewma.head(10)
利用长短期均线的相对关系,来识别价格趋势的变化。
Close2 = df.Close[-500:] # 重新取一个较大时间跨度的序列,500交易日数据
# 计算相关指标,用到了上边定义过的函数ewmaCal()。
DIF = ewmaCal(Close2, 12, 2/(1+12)) - ewmaCal(Close2, 26, 2/(1+26))
DEA = ewmaCal(DIF, 9, 2 / (1+9)) # DEA的线图也被称为信号线
MACD = DIF - DEA
# 提取2020年全年数据来作图
DIF = DIF['2020']
DEA = DEA['2020']
MACD = MACD['2020']
MACD1 = MACD[MACD.values > 0] # MACD数据大于零的部分 作图体现为红色
MACD2 = MACD[MACD.values < 0] # MACD数据小于零的部分 作图体现为绿色
# 绘制2020年全年深证成指MACD图像
plt.rcParams['font.sans-serif'] = ['SimHei'] # 设定显示中文
plt.rcParams['axes.unicode_minus'] = False # 设定正常显示负号
plt.subplot(211)
plt.plot(DIF, label="DIF", color="k")
plt.plot(DEA, label="DEA", color="b",linestyle='dashed')
plt.title("信号线DIF与DEA")
plt.xlabel('date')
plt.ylabel('value')
plt.legend()
plt.subplot(212)
plt.bar(MACD1.index, height=MACD1[MACD1.values > 0], label='MACD', color='r', edgecolor='k')
plt.bar(MACD2.index, height=MACD2[MACD2.values < 0], label='MACD', color='g', edgecolor='k')
plt.xlabel('date')
plt.ylabel('value')
plt.legend()
plt.show()
# plt.savefig(path + '/深证成指2020年MACD指标图.png')
这里我们阐述一些常用的MACD交易思想:
DIF与DEA的交叉与背离策略分析: