def LoadFile(self):
# get list
flist = tdx2.get_file_list()
self.msgManager.ShowMessage('total file count:'+ str(len(flist)))
flist = flist[:cm.getConfig('MAX_LOAD_ITEM')]
# filter_2
rflist = []
for k in flist:
if cm.getConfig('FILTER1'):
if k[:3] == "sh0": continue
if k[:3] == "sh1": continue
if k[:3] == "sh2": continue
if k[:3] == "sh3": continue
if k[:3] == "sh5": continue
if k[:3] == "sh8": continue
#if k[:3] == "sh5": continue
rflist.append(k)
flist = rflist
# read file main
gauage = 0
dlinelist = []
for k in flist:
dline = tdx2.get_dayline_by_fid(k, restrictSize = 40)
dlinelist.append([k,dline])
gauage+=1
if gauage%100 == 0: print gauage / 33, '%'
self.dlinelist = dlinelist
self.LoadProcess()
def LoadFile(self):
# get list
flist = tdx2.get_file_list()
self.msgManager.ShowMessage('total file count:' + str(len(flist)))
flist = flist[:cm.getConfig('MAX_LOAD_ITEM')]
# filter_2
rflist = []
for k in flist:
if cm.getConfig('FILTER1'):
if k[:3] == "sh0": continue
if k[:3] == "sh1": continue
if k[:3] == "sh2": continue
if k[:3] == "sh3": continue
if k[:3] == "sh5": continue
if k[:3] == "sh8": continue
#if k[:3] == "sh5": continue
rflist.append(k)
flist = rflist
# read file main
gauage = 0
dlinelist = []
for k in flist:
dline = tdx2.get_dayline_by_fid(k, restrictSize=40)
dlinelist.append([k, dline])
gauage += 1
if gauage % 100 == 0: print gauage / 33, '%'
self.dlinelist = dlinelist
self.LoadProcess()
def LoadFile(self):
flist = tdx2.get_file_list()
print 'total file count:', len(flist)
flist = flist[:MAX_LOAD_ITEM]
# filter 2
rflist = []
for k in flist:
if FILTER1:
if k[:3] == "sh0": continue
if k[:3] == "sh1": continue
if k[:3] == "sh2": continue
if k[:3] == "sh3": continue
if k[:3] == "sh5": continue
if k[:3] == "sh8": continue
#if k[:3] == "sh5": continue
rflist.append(k)
flist = rflist
datalist = []
dataItem = []
for k in flist:
dline = tdx2.get_dayline_by_fid(k, restrictSize=30)
if FILTER3:
if dline[-1].op > 60: continue
if len(dline) < 30: continue
print 'loading', k
dataItem = [k] + [k(dline) for k in varfunc]
datalist.append(dataItem)
self.CalculateLv2(datalist)
print 'data loading finished'
self.unfilteredList = datalist
def LoadFile(self):
flist = tdx2.get_file_list()
print 'total file count:', len(flist)
flist = flist[:MAX_LOAD_ITEM]
# filter 2
rflist = []
for k in flist:
if FILTER1:
if k[:3] == "sh0": continue
if k[:3] == "sh1": continue
if k[:3] == "sh2": continue
if k[:3] == "sh3": continue
if k[:3] == "sh5": continue
if k[:3] == "sh8": continue
#if k[:3] == "sh5": continue
rflist.append(k)
flist = rflist
datalist = []
dataItem = []
for k in flist:
dline = tdx2.get_dayline_by_fid(k, restrictSize = 30)
if FILTER3:
if dline[-1].op > 60: continue
if len(dline) < 30: continue
print 'loading', k
dataItem = [k] + [k(dline) for k in varfunc]
datalist.append(dataItem)
self.CalculateLv2(datalist)
print 'data loading finished'
self.unfilteredList = datalist
dline = tdx2.get_dayline_by_fid(k)
delta_rate = afunc.delta_rate(dline, 3)
dr = int(delta_rate * 10000) / 100.0
stlist.append([k, dr])
except:
pass
stlist.sort(key=lambda x: -x[1])
print "stlist generated"
return stlist
def write_to_file(stlist):
print "write to file start"
'''
Write k to file
k = ['sh000001', 1, 2, 3, 4, ....]
infile: sh00001 \t 1 \t 2 \t3 \t 4 \t \n ...
'''
f = open('period.txt', 'w')
for k in stlist:
f.write(k[0] + '\t')
for m in k[1:]:
f.write(str(m) + '\t')
f.write('\n')
print "write to file finished"
if __name__ == '__main__':
flist = tdx2.get_file_list()
stlist = generate_stlist(flist)
write_to_file(stlist)
# write function
try:
dline = tdx2.get_dayline_by_fid(k)
delta_rate = afunc.delta_rate(dline, 3)
dr = int(delta_rate * 10000) / 100.0
stlist.append([k, dr])
except:
pass
stlist.sort(key = lambda x: -x[1])
print "stlist generated"
return stlist
def write_to_file(stlist):
print "write to file start"
'''
Write k to file
k = ['sh000001', 1, 2, 3, 4, ....]
infile: sh00001 \t 1 \t 2 \t3 \t 4 \t \n ...
'''
f = open('period.txt', 'w')
for k in stlist:
f.write( k[0] + '\t')
for m in k[1:]:
f.write( str(m) + '\t')
f.write('\n')
print "write to file finished"
if __name__ == '__main__':
flist = tdx2.get_file_list()
stlist = generate_stlist(flist)
write_to_file(stlist)
