均線交易策略

目的：從所有股票中選出符合買入策略的股票。

符合買入條件：

1、當天5日均線數據大於等於10日均線數據；

2、昨天5日均線數據小於10日均線數據；

3、10日均線數據處於上升趨勢。

代碼實現如下：

import tushare as ts
import pandas as df
import datetime
def getStockHistoryData():
    data = ts.get_hist_data(code = "002456",start = "2019-07-20", end = "2019-7-26", ktype = "D")
    data.to_csv("stockData.csv")
    print(data)
    pass

def choiceStock():
    """
    :param stockData:
    :return:
    1、當天5日均線大於10日均線；
    2、前一天5日均線小於10日均線；
    """
    stockChoiced = []
    endDate = datetime.datetime.now()   #獲取當天日期
    offset = datetime.timedelta(days = -3)  #設置時間偏移量
    startDate = (endDate + offset).strftime("%Y-%m-%d")  #獲取startDate
    endDate = endDate.strftime("%Y-%m-%d")

    stockBasic = ts.get_stock_basics()        #獲取所有股票數據信息
    stockCode = stockBasic.index              #獲取所有股票代碼

    for i in range(1,30):
        code = stockCode[i]                    #
        getCondCode = stockChoicTragy(code,startDate,endDate)  #獲取滿足條件的股票信息
        if(getCondCode):
            stockChoiced.append(getCondCode)                   #將符合條件的股票加入到列表中

    return stockChoiced

def stockChoicTragy(code,startDate,endDate):
    data = ts.get_hist_data(code = code, start=startDate, end=endDate)  #獲取指定時間范圍內的股票數據
    data = data[["ma5","ma10","ma20"]].sort_index(ascending=False)   #按日期升序排列
    ma5Data = data["ma5"]       #獲取5日均線數據
    ma10Data = data["ma10"]     #獲取10日均線數據

    if(len(data) >= 2):
        condition1 = ma5Data[0] >= ma10Data[0]    #當天5日均線數據大於等於10日均線數據
        condition2 = ma5Data[1] < ma10Data[1]    #昨天5日均線數據小於10日均線數據
        condition3 = ma10Data[0] > ma10Data[1]   #10日均線數據呈上升趨勢

        if(condition1 and condition2 and condition3):
            return code

if __name__ == "__main__":
    stock = choiceStock()
    print(stock)