python通达信5分钟转,10分钟,15分钟,30分钟,60分钟,量化交易,K线

    import os

    import pandas as pd
    from pandas import Timedelta

    from stock_c.csv2dataframe import import_csv


    # 用通达信小周期,生成大周期数据
    def csv_resample(df, rule) -> pd.DataFrame:
        # 重新采样Open列数据
        df_open = round(df['Open'].resample(rule=rule, closed='right', label='left').first(), 2)
        df_high = round(df['High'].resample(rule=rule, closed='right', label='left').max(), 2)
        df_low = round(df['Low'].resample(rule=rule, closed='right', label='left').min(), 2)
        df_close = round(df['Close'].resample(rule=rule, closed='right', label='left').last(), 2)
        df_volume = round(df['Volume'].resample(rule=rule, closed='right', label='left').sum(), 2)
        # print("新周期数据已生成")
        # 生成新周期数据
        df_15t = pd.DataFrame()
        df_15t = df_15t.assign(Open=df_open)
        df_15t = df_15t.assign(High=df_high)
        df_15t = df_15t.assign(Low=df_low)
        df_15t = df_15t.assign(Close=df_close)
        df_15t = df_15t.assign(Volume=df_volume)
        # 去除空值
        df_15t = df_15t.dropna()

        return df_15t


    # 根据通达信5分钟周期数据,生成其他周期数据
    def lc5_resample(filepath, name, targetdir, rule) -> None:
        # (通达信.lc5文件路径, 通达信.lc5文件名称, 处理后要保存到的文件夹)
        # 设置处理后保存文件的路径和名称
        print("周期转换已开始: " + rule)
        file_object_path = targetdir + name.split('.')[0] + ".lc" + rule[:len(rule) - 1] + '.csv'
        df = import_csv(filepath)
        if rule == '60T':
            df = round(change_13_11_14_12(df), 2)

        df = csv_resample(df, rule)
        df.to_csv(file_object_path)
        print("数据转换已完成: " + name)


    def lc5_rule(rule):
        # 设置通达信5分钟周期数据文件所在的文件夹
        path_dir = '../lc5/'
        # 设置要转换的新周期
        rule_cycle = rule
        # 设置数据处理好后,要将csv文件保存的文件夹
        target_dir = '../lc' + rule_cycle[:len(rule_cycle) - 1] + '/'
        # 读取文件夹下的通达信.lc5.csv文件
        listfile = os.listdir(path_dir)
        # 逐个处理文件夹下的通达信.lc5.csv文件,并生成对应的csv文件,保存到对应周期文件夹下
        for fname in listfile:
            lc5_resample(path_dir + fname, fname, target_dir, rule_cycle)
        else:
            print('The for ' + path_dir + ' to ' + target_dir + '  loop is over')
            print("文件转换已完成")


    def change_13_11_14_12(df) -> pd.DataFrame:
        date = []
        for i in df.index:
            if i.hour == 13:
                i = i - Timedelta('02:00:00')
            if i.hour == 14 and i.minute == 0 and i.second == 0:
                i = i - Timedelta('02:00:00')
            date.append(i)
        df = df.assign(Date=pd.Series(date, index=df.index))
        df.set_index(['Date'], inplace=True)
        return df


    # 转换成新周期
    lc5_rule('10T')
    lc5_rule('15T')
    lc5_rule('30T')
    lc5_rule('60T')


    # 读取csv文件,返回pd.DataFrame对象
    def import_csv(stock_code) -> pd.DataFrame:
        df = pd.read_csv(stock_code)
        df['Date'] = pd.to_datetime(df['Date'], format='%Y/%m/%d')
        df.set_index(['Date'], inplace=True)
        return df

  • 2
    点赞
  • 29
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 1
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

一方天地006

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值