零基础搭建量化投资系统—以Python为基础:视频课件及代码(3)

零基础搭建量化投资系统—以Python为基础:视频课件及代码(3)
自编股票指标公式及显示

import  tkinter  as  tk   #导入Tkinter
import  tkinter.ttk  as  ttk   #导入Tkinter.ttk
import  tkinter.tix  as  tix   #导入Tkinter.tix
import  HP_tk  as  htk   #导入htk
from PIL import Image, ImageTk, ImageDraw, ImageFont
from matplotlib import ticker as mticker
import time
import  HP_view as hpv
import matplotlib.pyplot as plt
from HP_formula import *    #使用HP_formula公式基本函数库
#建立主窗口
root=htk.MainWindow(title='自定义公式演示',x=100,y=200,w=800, h=600)
root.iconbitmap('ico/py.ico')   #设置应用程序图标
root.SetCenter()                         #移动到屏幕中央
#把window划分为左右2个子容器,左为v1,右为v2
v1=tk.Frame(root,width = 200,bg='blue')
v1.pack(side=tk.LEFT,fill=tk.Y)
v2=tk.Frame(root,bg='yellow')
v2.pack(side=tk.RIGHT, fill=tk.BOTH, expand=1)
#《零基础搭建量化投资系统—以Python为基础》树中示例 8-23
tree=ttk.Treeview(v1)
myid=tree.insert("",0,"指标",text="指标",values=("1"),open=True)
myidx1=tree.insert(myid,0,"技术指标",text="技术指标",values=("2"))
myidx2=tree.insert(myid,1,"交易系统",text="交易系统",values=("3"))
myidx3=tree.insert(myid,2,"条件选股",text="条件选股",values=("4"))
myidx4=tree.insert(myid,3,"五彩 K 线",text="五彩 K 线",values=("5"))
myidy=tree.insert("",1,"工具",text="工具",values=("6"),open=True)
myidy1=tree.insert(myidy,0,"系统设置",text="系统设置",values=("7"))
myidy1=tree.insert(myidy,0,"指标排序",text="指标排序",values=("8"))
tree.pack(side=tk.TOP,fill=tk.BOTH, expand=1)
tree.selection_set("指标")

ds='2018-01-01'  #行情开始日期
de=time.strftime('%Y-%m-%d',time.localtime(time.time()))  #行情结束日期
###---------------------------------------------------------------------------
#本部分四选一
#二. #小白离线行情,需要xbdata数据
import tushare as ts    #小白离线行情,需要xbdata数据
#读取股票数据
df2a=ts.get_k_data('300202',ktype='D',start=ds,end=de,index=False,autype='qfq')

#六均线3指标图,K线算一个指标
ax,ax2=hpv.axview3x2(v2,df2a,'300202聚龙股份  六均线K线演示',6)  

#在ax区绘制荷蒲操盘线HPCPX
def draw_HPCPX(ax1,mydf):
    CLOSE=mydf['close']
    LOW=mydf['low']
    HIGH=mydf['high']
    OPEN=mydf['open']
    VOL=mydf['volume']
    df=mydf.copy()

    def HPCPX():
        S0=(CLOSE+OPEN+LOW+HIGH)/4
        S1=EMA(SLOPE(S0,5),60)
        S2=EMA(S1,20)
        S3=IF(S1>=0,S1,DRAWNULL())
        S4=IF(S1>S2 ,S3,DRAWNULL())
        return S1,S2,S4
    #使用KDJ指标，返回K，D，J序列。
    x,y,z=HPCPX()
    
    df = df.join(pd.Series( x,name='CPX'))
    df = df.join(pd.Series( y,name='Y'))
    df = df.join(pd.Series( z,name='Z'))
    df['Y0']=0
    
    
    #开始绘图
    ax1.plot(df.date.values,df.CPX.values, color= 'green', lw=2,label="荷蒲操盘线")
    ax1.plot(df.date.values, df.Z.values, color= 'red', lw=2)
    ax1.plot(df.date.values, df.Y0.values, color= 'yellow', lw=2)
    text = ax1.text(0.05, 0.05, '红线持股！绿线持币！', fontdict={'size': 20},color='yellow')
    plt.ylabel('HPCPW', color='white')
    plt.legend() # 显示图中右上角的提示信息。
    plt.gca().yaxis.set_major_locator(mticker.MaxNLocator(prune='upper'))
    ax1.xaxis.set_major_locator(mticker.MaxNLocator(8))  #x轴分成几等分     
    ax1.yaxis.set_major_locator(mticker.MaxNLocator(nbins=3, prune='upper'))
    ax1.grid(True, color='r')
    ax1.tick_params(axis='x', colors='white')
    ax1.tick_params(axis='y', colors='white')
    plt.subplots_adjust(left=.075, bottom=.08, right=.96, top=.96, wspace=.15, hspace=0.1)
    plt.close() # 关窗口
    return ax1

#调用自定义公式
ax3=draw_HPCPX(ax2,df2a)  
root.mainloop()  	#进入Tkinter消息循环