TomDeMark9.py

from common.framework import *

resultgt_list = []
resultlt_list = []
#greater than 大于
def isgt(dts):
    i1 = dts.index[-1]
    #i2 = dts.index[-2]
    #i3 = dts.index[-3]
    #i4 = dts.index[-4]
    i5 = dts.index[-5]

    if (dts.close[i1] > dts.close[i5]
       #and 
       #dts.close[i1] > dts.close[i3]
       # and 
       #dts.close[i1] > dts.close[i4]
       # and 
       #dts.close[i1] > dts.close[i5]
       ):
        return True
    else:
        return False

#less than 小于
def islt(dts):
    i1 = dts.index[-1]
    #i2 = dts.index[-2]
    #i3 = dts.index[-3]
    #i4 = dts.index[-4]
    i5 = dts.index[-5]

    if (dts.close[i1] < dts.close[i5]
       # and
       #dts.close[i1] < dts.close[i3]
       # and
       #dts.close[i1] < dts.close[i4]
       # and
       #dts.close[i1] < dts.close[i5]):
       ):
        return True
    else:
        return False


def td9(code,name,datas):
    print(code,name)
    gtstatus = 0
    ltstatus = 0
    if len(datas)<7:
        return
    for i in range(5,len(datas)):
        if isgt(datas[i-4:i+1]):
            gtstatus += 1
            if gtstatus > 3 and i == (len(datas)-1):
                turn = datas.turn[datas.index[i]]
                volume = datas.volume[datas.index[i]]
                if volume < 1 or float(turn) == 0 :
                    continue
                hqltsz = float(datas.close[datas.index[i]]) * float(volume) / float(turn) / 1000000 
                hqltsz = float('%.2f' % hqltsz)
                if hqltsz < 50.0:
                    continue
                print(OKRED,datas.date[datas.index[i]],gtstatus,turn,volume,hqltsz,ENDC)
                resultgt_list.append([name,code,datas.date[datas.index[i]],gtstatus,hqltsz])
        else:
            gtstatus = 0

        if islt(datas[i-4:i+1]):
            ltstatus += 1
            if ltstatus > 3 and i == (len(datas)-1):
                turn = datas.turn[datas.index[i]]
                volume = datas.volume[datas.index[i]]
                if volume < 1 or float(turn) ==0 :
                    continue
                hqltsz = float(datas.close[datas.index[i]]) * float(volume) / float(turn) / 1000000 
                hqltsz = float('%.2f' % hqltsz)
                if hqltsz < 50.0:
                    continue
                print(OKGREEN,datas.date[datas.index[i]],ltstatus,turn,volume,hqltsz,ENDC)
                resultlt_list.append([name,code,datas.date[datas.index[i]],ltstatus,hqltsz])
    
        else:
            ltstatus = 0    
def display():
    for i in resultgt_list  + resultlt_list:
        print(i)

def save():
        df = pd.DataFrame(resultgt_list +[['0','sh.0000','0',1,10.0]]+ resultlt_list, columns = ['name','code','date','9转第N天','流通股值'])
        print("保存在",'./datas/stock_'+endday+"9dt.html")
        save_df_tohtml('./datas/stock_'+endday+"9dt.html",df)

if __name__ == "__main__":
    init_stock_list()
    loop_all(td9)
    display()
    save()