大家好,又见面了,我是你们的朋友全栈君。
使用Python激活成功教程通达信股票数据
最近在研究软件选股与自动交易,所以需要获取股票的每日数据。
我的电脑上已经安装了券商的交易软件,交易软件提供商是通达信。
下面简要介绍从通达信的交易软件中获取股票数据的过程,使用Python语言。
本文参考了CSDN的一篇文章,原始链接如下:
http://blog.csdn.net/liuyukuan/article/details/53560278
一、通达信的股票文件格式
通过交易软件的“盘后数据下载”功能,可以下载到所有股票的日线数据,具体位置在:
交易软件的安装目录vipdocsh[sz]lday
文件命名为:sh[sz]000001.day
日线文件中,每32字节(32bytes)为一天的记录。
每4个字节为一项数据:
第1项,交易日期
第2项,开盘价
第3项,最高价
第4项,最低价
第5项,收盘价
第6项,成交金额
第7项,成交量
第8项,未使用
二、Python代码
1. 设计思路:
将日线文件转换成CS格式的文件,方便后期处理。
CSV是一种文本文件的格式,使用逗号或其它分隔符将数据逐列隔开,每行代表一条记录。
可以使用Excel打开CSV格式的文件。
打开日线文件,依次读入32个字节,然后将这8项数据变成字符格式后以逗号隔开,写入CSV文件。
2. 主要函数介绍:
这里用到了strut库中的unpack函数,该函数用于将字节序列解析为字符串,有许多可用的格式。
3. 代码:
代码语言:javascript复制#!/usr/bin/python
# -*- coding: UTF-8 -*-
import os
from struct import unpack
# 将通达信的日线文件转换成CSV格式
def day2csv(source_dir, file_name, target_dir):
# 以二进制方式打开源文件
source_file = open(source_dir os.sep file_name, 'rb')
buf = source_file.read()
source_file.close()
# 打开目标文件,后缀名为CSV
target_file = open(target_dir os.sep file_name '.csv', 'w')
buf_size = len(buf)
rec_count = buf_size / 32
begin = 0
end = 32
header = str('date') ', ' str('open') ', ' str('high') ', ' str('low') ', '
str('close') ', ' str('amount') ', ' str('vol') ', ' str('str07') 'n'
target_file.write(header)
for i in xrange(rec_count):
# 将字节流转换成Python数据格式
# I: unsigned int
# f: float
a = unpack('IIIIIfII', buf[begin:end])
line = str(a[0]) ', ' str(a[1] / 100.0) ', ' str(a[2] / 100.0) ', '
str(a[3] / 100.0) ', ' str(a[4] / 100.0) ', ' str(a[5] / 10.0) ', '
str(a[6]) ', ' str(a[7]) ', ' 'n'
target_file.write(line)
begin = 32
end = 32
target_file.close()
source = 'C:/new_dxzq_v6/vipdoc/sh/lday'
target = 'E:/tmp'
file_list = os.listdir(source)
for f in file_list:
day2csv(source, f, target)
发布者:全栈程序员栈长,转载请注明出处:https://javaforall.cn/149090.html原文链接:https://javaforall.cn