自己做量化交易软件(20)通达信公式选股程序的实现
1.用小白2很好实现。首先获取板块中股票,存放在列表中。例如获取"上证50"
import HP_tdx as htdxfrom HP_formula import *global CLOSE,LOW,HIGH,OPEN,VOLglobal C,L,H,O,Vhq=htdx.TdxInit(ip='183.60.224.178',port=7709) ##初始化通达信codes=htdx.getblock2('上证50') #返回板块中的股票
2.下来用小白仿通达信公式模式进行指标计算,存放到B和S列表中。
如果是买点,B列表判断最后一个数据是否为1。如果为1,就保存股票代码到选股池中,这个就是选股算法。
因为股票行情数据太多,计算比较慢,不要获取全部数据,如果你用MA5上穿MA20,因为使用了最长参数20, 所以你要至少获取40个周期的数据,才能准确。
如果用macd,最长周期参数26,你至少要获取52个周期数据行情。其他以此类推。即保证了指标计算的准确,又提高了计算速度。
我们把这个选股做成函数function(cd),参数cd表示股票代码。
选股策略’5日均线上穿20日均线’,下面给出实现函数。
#均线选股函数:5日均线上穿20日均线def function(cd):global CLOSE,LOW,HIGH,OPEN,VOLglobal C,L,H,O,Vdf3=htdx.get_security_bars(nCategory=4,nMarket = 0,code=cd,\nStart=0, nCount=40) #获取指定范围的证券K线##数据规格化 df3.dropna(inplace=True)#小白数据规格化mydf=df3.copy()CLOSE=mydf['close']LOW=mydf['low']HIGH=mydf['high']OPEN=mydf['open']VOL=mydf['volume']C=mydf['close']L=mydf['low']H=mydf['high']O=mydf['open']V=mydf['volume']#仿通达信,大智慧公式计算和选股MA5=MA(CLOSE,5)MA20=MA(CLOSE,20)B=CROSS(MA5,MA20) #5日均线上穿20日均线B2=list(B) #把1维数组转换为列表数据return B2[len(B2)-1] #获取列表中最后一个值
3.然后循环获取股票池列表中的股票行情,并进行计算。
#下面开始进行板块或股票池选股myblock=[] #选股池for code in codes:buy=function(code) #计算股票选股函数if buy==1:myblock.append(code)
4.输出结果
print('输出选出的股票池')print(myblock)
5.下面是选出结果:
小白量化选股: 5日均线上传20日均线输出股票池或板块中股票['600000', '600009', '600016', '600028', '600030', '600031', '600036', '600048', '600050', '600104', '600196', '600276', '600309', '600340', '600519', '600547', '600585', '600690', '600703', '600837', '600887', '601012', '601066', '601088', '601111', '601138', '601166', '601186', '601211', '601236', '601288', '601318', '601319', '601328', '601336', '601390', '601398', '601601', '601628', '601668', '601688', '601766', '601818', '601857', '601888', '601939', '601988', '601989', '603259', '603993']输出选出的股票池['600000', '600690']
下面给出全部代码.
#小白量化选股 : 5日均线上穿20日均线#购买<零基础搭建量化投资系统>正版书,送小白量化软件源代码。# /61567375505.html#独狼荷蒲qq:2775205#通通python量化群:524949939#电话微信:18578755056#微信公众号:独狼股票分析import HP_tdx as htdxfrom HP_formula import *global CLOSE,LOW,HIGH,OPEN,VOLglobal C,L,H,O,Vhq=htdx.TdxInit(ip='183.60.224.178',port=7709) ##初始化通达信codes=htdx.getblock2('上证50') #返回板块中的股票print('小白量化选股: 5日均线上传20日均线')print('输出股票池或板块中股票')print(codes)#均线选股函数def function(cd):global CLOSE,LOW,HIGH,OPEN,VOLglobal C,L,H,O,Vdf3=htdx.get_security_bars(nCategory=4,nMarket = 0,code=cd,\nStart=0, nCount=40) #获取指定范围的证券K线##数据规格化 df3.dropna(inplace=True)#小白数据规格化mydf=df3.copy()CLOSE=mydf['close']LOW=mydf['low']HIGH=mydf['high']OPEN=mydf['open']VOL=mydf['volume']C=mydf['close']L=mydf['low']H=mydf['high']O=mydf['open']V=mydf['volume']#仿通达信,大智慧公式计算和选股MA5=MA(CLOSE,5)MA20=MA(CLOSE,20)B=CROSS(MA5,MA20) #5日均线上穿20日均线B2=list(B)return B2[len(B2)-1]#下面开始进行板块或股票池选股myblock=[] #选股池for code in codes:buy=function(code) #计算股票选股函数if buy==1:myblock.append(code)print('输出选出的股票池')print(myblock)
读者可以根据上面的代码,改为自己的自编指标进行选股.
京东、淘宝、当当网等购买正版<零基础搭建量化投资系统——以Python为工具>书,加读者群,除了获得小白量化源代码外,赠送第2代小白量化源代码和相关金融模块源代码。HP_tdx.py是第2代中实时行情模块。