通达信 python插件选股_自己做量化交易软件(20)通达信公式选股程序的实现
自己做量化交易软件(20)通达信公式选股程序的实现
1.用小白2很好实现。首先获取板块中股票，存放在列表中。例如获取"上证50"
import HP_tdx as htdx
from HP_formula import *
global CLOSE,LOW,HIGH,OPEN,VOL
global C,L,H,O,V
hq=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,VOL
global C,L,H,O,V
df3=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.下面是20200411选出结果:
小白量化选股: 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日均线
#购买<零基础搭建量化投资系统>正版书,送小白量化软件源代码。
# https://item.jd.com/61567375505.html
#独狼荷蒲qq:2775205
#通通python量化群:524949939
#电话微信:18578755056
#微信公众号：独狼股票分析
import HP_tdx as htdx
from HP_formula import *
global CLOSE,LOW,HIGH,OPEN,VOL
global C,L,H,O,V
hq=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,VOL
global C,L,H,O,V
df3=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代中实时行情模块。
原文链接:https://blog.csdn.net/hepu8/article/details/105451351