[python]沪深龙虎榜数据导入通达信的自选板块，并标注于K线图上
将沪深龙虎榜数据导入通达信的自选板块，并标注于K线图上原理：python读取前一次处理完的计算5日后涨跌幅输出的csv文件
文件名前加""
安照通达信的画图文件和板块文件格式，输出文件
用通达信的导入功能，导入画图文件和板块文件即可
事前数据截图：
处理后所得文件：
导入通达信后板块截图：
k线截图：
代码：
1 #coding=utf-8
2
3 #读取''开头的csv文件
4 #copyright @ WangXinsheng
5 #http://www.cnblogs.com/wangxinsheng/
6 import os
7 import struct
8 import time
9 import datetime
10
11 def getPriceByte(infile,arg,code,ss,day):
12 dirc = {"-5":-3,"-3":-2,"-1":-1,"1":1,"3":2,"5":3,"7":4,"9":5,"11":6}
13 first = 0
14 second = 0
15 a = float(arg)
16 for i in [-5,-3,-1,1,3,5,7,9,11]:
17 ib = i - 2
18 if(a>=2**ib and a<2**i):
19 #print(str(a)+":["+str(2**ib)+","+str(2**i)+"]"+"i:"+str(i)+" ib:"+str(ib))
20 first = ib
21 break;
22 #print(first)
23 chushu = 2**first
24 if(arg >= chushu*2):
25 second = arg / chushu
26 #print(second)
27 second = int(second*64)
28 else:
29 second = arg / chushu -1
30 #print(second)
31 second = int(second*128)
32
33 if(first<0):
34 first = 64 + dirc
35 else:
36 first = 63 + dirc
37 '''
38 print(first)
39 print(second)
40 print(struct.pack('B',first))
41 print(struct.pack('B',second))
42 '''
43 #print(struct.pack('B',first))
44 #print(struct.pack('B',second))
45
46 infile.write(struct.pack('B',ss))
47 infile.write(code.encode())
48 infile.write(b'\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00')
49 infile.write(b'\x03')
50 infile.write('STANDK'.encode())
51 infile.write(b'\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00')
52 infile.write(b'\x04')
53 infile.write(b'\x00')
54 infile.write(b'\x1a\x10\x00\x00')
55 infile.write(struct.pack('i',int(day)))
56 infile.write(b'\x00\x00')
57 infile.write(struct.pack('B',second))
58 infile.write(struct.pack('B',first))
59 infile.write(b'\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\xdcz3\x01\x01\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00')
60
61 print(str(code)+':已在 '+day+"日，价格："+str(arg)+"加入标记")
62 #infile = open("tdxline.eld","rw")
63 #getPriceByte(5.55)
64 file = open("WXSline.eld","wb")
65 zxg = ''
66 path=r'.'
67 files = os.listdir(path)
68 files.sort()
69 j=0
70 for f in files:
71 if(f.startswith('') and f.endswith('.csv')):
72 print('读取文件：'+path+'/'+f)
73
74 f=open(path+'/'+f,'rt')
75 infos = f.readlines()
76 f.close()
77 i=0
78 for info in infos:
79 if(i==0):
80 i=i+1
81 continue
82 tmp = info.split(',')
83 if(len(tmp)<2):
84 continue
85 code = str(tmp.replace('"','').replace("'",""))
86 if(code.startswith('6')):
87 ss = 1
88 else:
89 ss = 0
90 addzxg = ''
91 try:
92 zxg.index(str(ss)+code)
93 except:
94 addzxg = str(ss)+code+"\n"
95
96 if(addzxg!=''):
97 zxg = zxg + str(ss)+code+"\n"
98 if(len(tmp)<10):
99 continue
100
101 days = tmp.split('-')
102 if(len(days)<2):
103 days = tmp.split('/')
104
105 if(len(days)<2):
106 continue
107
108 if(len(days)<2):
109 days = '0'+days
110 if(len(days)<2):
111 days = '0'+days
112 day = days+days+days
113 #print(day)
114
115 priceArr = tmp.replace('"','').split('%')
116 if(len(priceArr)<2):
117 continue
118 a = float(priceArr.replace('[','').replace(']',''))*(1-float(priceArr.replace('%',''))/100)
119 a = round(a,2)
120 code = str(tmp.replace('"','').replace("'",""))
121 if(code.startswith('6')):
122 ss = 1
123 else:
124 ss = 0
125
126 getPriceByte(file,a,code,ss,day)
127 '''
128 if(j>3):
129 break
130 else:
131 j=j+1
132 '''
133
134 fileA = open("WXS条件股.ebk","w")
135 fileA.write(zxg)
136 fileA.close()
137 file.close()
138 print('处理完了\n画图文件地址：./WXSline.eld')
139 print('\n板块文件地址：./WXS条件股.ebk')
页:
[1]