零基础搭建量化投资系统—以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消息循环
