sma策略（用tushare包）

Question

1. 简易版SMA策略一旦遇到股价震荡频率较高的股票，收益往往会出现问题，因此做个简单的策略升级。基于简易版，设置股价差20为阈值。当出现金叉且股价差大于20时再做多，当出现死叉且股价差小于20时再做空。分别计算大盘和策略的累积收益，并可视化为折线图，并与简易版作对比，看策略收益存在多大的区别
2. 使用动量交易策略分析，分别以自身（本身/2天/6天）的平均return作为position，生成strategy再分别做可视化折线图
   简易版为下图 （回答附上code 感谢）

Answer 1

在原来计算出移动平均数的基础上，增加一些判断条件进行计算并画出对比图。参考如下代码：

#无策略
df['date_rev'] = [0]+[(df.loc[i,'close']-df.loc[i-1,'close'])
                      for i in range(1, len(df))]
df['ac_rev'] = df['date_rev'].cumsum()
#第一策略
a=[0]
for i in range(1, len(df)):
    if df.loc[i-1, 'm10'] < df.loc[i-1, 'm60'] and df.loc[i, 'm10'] > df.loc[i, 'm60']:
        n = i
    if df.loc[i-1, 'm10'] > df.loc[i-1, 'm60'] and df.loc[i, 'm10'] < df.loc[i, 'm60']:
        a.append(df.loc[i, 'close']-df.loc[n, 'close'])
    else:
        a.append(0)
df['st1_rev']=a
df['ac_st1_rev']=df['st1_rev'].cumsum()
#第二策略
x = [0]
for i in range(1,len(df)):
    if df.loc[i-1, 'm10'] < df.loc[i-1, 'm60'] and df.loc[i, 'm10'] > df.loc[i, 'm60'] and abs(df.loc[i,'close']-df.loc[i-1,'close']) >= 20:
        n = i
    if df.loc[i-1, 'm10'] > df.loc[i-1, 'm60'] and df.loc[i, 'm10'] < df.loc[i, 'm60'] and abs(df.loc[i, 'close']-df.loc[i-1, 'close']) >= 20:
        x.append(df.loc[i, 'close']-df.loc[n, 'close'])
    else:
        x.append(0)
#绘制图
df['st2_rev'] = x
df['ac_st2_rev'] = df['st2_rev'].cumsum()
df.to_csv('hs300_2.csv')
plt.figure(figsize=(12,8))
plt.subplot(1,2,1)
plt.title('strategy1')
plt.plot(df['date'], df['ac_rev'],df['ac_st1_rev'])
plt.legend(['ac_rev','ac_st1_rev'])
plt.subplot(1,2,2)
plt.title('strategy2')
plt.plot(df['date'], df['ac_rev'],df['ac_st2_rev'])
plt.legend(['ac_rev', 'ac_st2_rev'])
plt.show()