标签：文件 python str csv open day stock

import os
import struct
import datetime


# 读取通达信.day文件，并生成对应名称的csv文件
def stock_csv(filepath, name, targetdir) -> None:
    # (通达信.day文件路径, 通达信.day文件名称, 处理后要保存到的文件夹)

    with open(filepath, 'rb') as f:  # 读取通达信.day文件，并处理
        file_object_path = targetdir + name + '.csv'  # 设置处理后保存文件的路径和名称
        file_object = open(file_object_path, 'w+')  # 打开新建的csv文件，开始写入数据
        title_str = "Date,Open,High,Low,Close,Open_interest,Volume,settlement_price\n"  # 定义csv文件标题
        file_object.writelines(title_str)  # 将文件标题写入到csv中
        while True:
            stock_date = f.read(4)  # 读取0-3
            stock_open = f.read(4)  # 读取4-7
            stock_high = f.read(4)  # 读取8-11
            stock_low = f.read(4)  # 读取12-15
            stock_close = f.read(4)  # 读取16-19
            stock_open_interest = f.read(4)  # 读取20-23
            stock_vol = f.read(4)  # 读取24-27
            stock_settlement_price = f.read(4)  # 读取28-31

            # date,open,high,low,close,open_interest,vol,settlement_price

            if not stock_date:
                break
            stock_date = struct.unpack('l', stock_date)  # 4字节 如20091229
            stock_open = struct.unpack('f', stock_open)  # 开盘价
            stock_high = struct.unpack('f', stock_high)  # 最高价
            stock_low = struct.unpack('f', stock_low)  # 最低价
            stock_close = struct.unpack('f', stock_close)  # 收盘价
            stock_open_interest = struct.unpack('l', stock_open_interest)  # 持仓量
            stock_vol = struct.unpack('l', stock_vol)  # 成交量
            stock_settlement_price = struct.unpack("f", stock_settlement_price)  # 结算价

            date_format = datetime.datetime.strptime(str(stock_date[0]), '%Y%M%d')  # 格式化日期
            day_str = date_format.strftime('%Y-%M-%d') + "," + str(stock_open[0]) + "," + str(stock_high[0]) + "," \
                      + str(stock_low[0]) + "," + str(stock_close[0]) + "," + str(stock_open_interest[0]) + "," \
                      + str(stock_vol[0]) + "," + str(stock_settlement_price[0]) + "\n"
            file_object.writelines(day_str)  # 将文件写入到csv文件中
        file_object.close()     # 完成数据写入


# 设置通达信.day文件所在的文件夹
path_dir = 'C:\\new_tdxqh\\vipdoc\\ds\\lday\\'
# 设置数据处理好后，要将csv文件保存的文件夹
target_dir = '../day/'
# 读取文件夹下的通达信.day文件
listfile = os.listdir(path_dir)
# 逐个处理文件夹下的通达信.day文件，并生成对应的csv文件，保存到../day/文件夹下
for fname in listfile:
    stock_csv(path_dir + fname, fname, target_dir)
else:
    print('The for ' + path_dir + ' to ' + target_dir + '  loop is over')
    print("文件转换已完成")

标签：文件,python,str,csv,open,day,stock
来源： https://blog.csdn.net/qq_38242953/article/details/120083856

本站声明： 1. iCode9 技术分享网（下文简称本站）提供的所有内容，仅供技术学习、探讨和分享；
2. 关于本站的所有留言、评论、转载及引用，纯属内容发起人的个人观点，与本站观点和立场无关；
3. 关于本站的所有言论和文字，纯属内容发起人的个人观点，与本站观点和立场无关；
4. 本站文章均是网友提供，不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属；如您发现该文章侵犯了您的权益，可联系我们第一时间进行删除；
5. 本站为非盈利性的个人网站，所有内容不会用来进行牟利，也不会利用任何形式的广告来间接获益，纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。