金融分析之基本ma5 及金叉、死叉

import pandas as pd
import numpy as np
import matplotlib.pylab as plt

res=pd.read_csv("601318.csv",parse_dates=True,index_col="date")


#画图
# plt.plot(res["open"],"r-",res["close"],"b-")
# plt.xlabel(u"time")
# plt.ylabel(u"price")
# plt.show()

#计算ma5和ma10

#----------方案1
# res["ma5"]=np.NaN
# res["ma10"]=np.NaN
#
# for i in range(4,len(res)):       #从4开始
#     res.loc[res.index[i],"ma5"]=res["close"][i-4:i+1].mean()  #按下表取到时间对象，然后loc拿到该位置的ma5值，然后取到平均值
#
# for i in range(9,len(res)):
#     res.loc[res.index[i],"ma5"]=res["close"][i-9:i+1].mean()
#
# print(res)

#----------方案2
# res["ma5"]=(res["close"].cumsum()-res["close"].cumsum().shift(1).fillna(0).shift(4))/5      #如果一次性移动5个的话，第五个
#位置会成为NaN，这样，第五个位置上相减后就会没有值，所以先移动一个，让其填充上0，这样保证前四个是NaN，但是第五个有值
# res["ma10"]=(res["close"].cumsum()-res["close"].cumsum().shift(1).fillna(0).shift(9))/10


# 1   2   3   4   5   6   7
# 1   3   6   10  15  21  28
#                 1   3   6   10

#----------方案3
res["ma5"]=res["close"].rolling(5).mean()     #默认rolling的光标行是最后一行
res["ma10"]=res["close"].rolling(10).mean()

# plt.plot(res["ma5"],"r-",res["ma10"],"b-")
# plt.xlabel(u"time")
# plt.ylabel(u"price")
# plt.show()


#找出金叉和死叉

res=res.dropna()

#----------------方案1
# golden=[]
# black=[]
# sr=res["ma5"]>=res["ma10"]
#
# for i in range(1,len(sr)):
#     if sr.iloc[i]==True and sr.iloc[i-1]==False:
#         golden.append(sr.index[i])
#     if sr.iloc[i]==False and sr.iloc[i-1]==True:
#         black.append(sr.index[i])

#----------------方案2
golden=res[list((res["ma5"]>=res["ma10"]) & (res["ma5"]<res["ma10"]).shift(1))].index

black=res[(res["ma5"]<=res["ma10"]) & (res["ma5"]>res["ma10"]).shift(1)].index

# print(type((res["ma5"]>=res["ma10"]) & (res["ma5"]<res["ma10"]).shift(1)))
# print(list((res["ma5"]>=res["ma10"]) & (res["ma5"]<res["ma10"]).shift(1)))

#1、先取到5日大于10日的True，在取到5日小于10日的True，让后者往后挪一天则按照布尔索引取到所有值，然后再转化为索引

#                    1     2   3   4   5   6   7   8   9   10  11  12  13  14  15
#例如：5日>10日：#     T     T   T   F   T   T   T   T   F   F   F   F   T   T   T
#例如：5日<10日：#     F     F   F   T   F   F   F   F   T   T   T   T   F   F   F
#例如：5日<10日，后挪1：#     F   F   F   T   F   F   F   F   T   T   T   T   F   F   F

#死叉是4日和9日，则都是T的时候是金叉，如5日