1 import pandas as pd
 2 import numpy as np
 3 import matplotlib.pylab as plt
 4
 5 res=pd.read_csv("601318.csv",parse_dates=True,index_col="date")
 6
 7
 8 #画图
 9 # plt.plot(res["open"],"r-",res["close"],"b-")
10 # plt.xlabel(u"time")
11 # plt.ylabel(u"price")
12 # plt.show()
13
14 #计算ma5和ma10
15
16 #----------方案1
17 # res["ma5"]=np.NaN
18 # res["ma10"]=np.NaN
19 #
20 # for i in range(4,len(res)):       #从4开始
21 #     res.loc[res.index[i],"ma5"]=res["close"][i-4:i+1].mean()  #按下表取到时间对象，然后loc拿到该位置的ma5值，然后取到平均值
22 #
23 # for i in range(9,len(res)):
24 #     res.loc[res.index[i],"ma5"]=res["close"][i-9:i+1].mean()
25 #
26 # print(res)
27
28 #----------方案2
29 # res["ma5"]=(res["close"].cumsum()-res["close"].cumsum().shift(1).fillna(0).shift(4))/5      #如果一次性移动5个的话，第五个
30 #位置会成为NaN，这样，第五个位置上相减后就会没有值，所以先移动一个，让其填充上0，这样保证前四个是NaN，但是第五个有值
31 # res["ma10"]=(res["close"].cumsum()-res["close"].cumsum().shift(1).fillna(0).shift(9))/10
32
33
34 # 1   2   3   4   5   6   7
35 # 1   3   6   10  15  21  28
36 #                 1   3   6   10
37
38 #----------方案3
39 res["ma5"]=res["close"].rolling(5).mean()     #默认rolling的光标行是最后一行
40 res["ma10"]=res["close"].rolling(10).mean()
41
42 # plt.plot(res["ma5"],"r-",res["ma10"],"b-")
43 # plt.xlabel(u"time")
44 # plt.ylabel(u"price")
45 # plt.show()
46
47
48 #找出金叉和死叉
49
50 res=res.dropna()
51
52 #----------------方案1
53 # golden=[]
54 # black=[]
55 # sr=res["ma5"]>=res["ma10"]
56 #
57 # for i in range(1,len(sr)):
58 #     if sr.iloc[i]==True and sr.iloc[i-1]==False:
59 #         golden.append(sr.index[i])
60 #     if sr.iloc[i]==False and sr.iloc[i-1]==True:
61 #         black.append(sr.index[i])
62
63 #----------------方案2
64 golden=res[list((res["ma5"]>=res["ma10"]) & (res["ma5"]<res["ma10"]).shift(1))].index
65
66 black=res[(res["ma5"]<=res["ma10"]) & (res["ma5"]>res["ma10"]).shift(1)].index
67
68 # print(type((res["ma5"]>=res["ma10"]) & (res["ma5"]<res["ma10"]).shift(1)))
69 # print(list((res["ma5"]>=res["ma10"]) & (res["ma5"]<res["ma10"]).shift(1)))
70
71 #1、先取到5日大于10日的True，在取到5日小于10日的True，让后者往后挪一天则按照布尔索引取到所有值，然后再转化为索引
72
73 #                    1     2   3   4   5   6   7   8   9   10  11  12  13  14  15
74 #例如：5日>10日：#     T     T   T   F   T   T   T   T   F   F   F   F   T   T   T
75 #例如：5日<10日：#     F     F   F   T   F   F   F   F   T   T   T   T   F   F   F
76 #例如：5日<10日，后挪1：#     F   F   F   T   F   F   F   F   T   T   T   T   F   F   F
77
78 #死叉是4日和9日，则都是T的时候是金叉，如5日