Function file_split

tools/tools.py:198–232

(input_file, num)

        print(f"处理文件时出现错误：{e}")

def file_split(input_file, num):
        # 假设你的文件名是 'data.txt'
        # input_file = 'input_file.txt'
        input_name = input_file.split('.')[0]
        file_type = input_file.split('.')[1]
        output_files = []
        for i in range(num):
            output_files.append(f'{input_name}_part{i}.{file_type}')

        # 计算文件总行数
        with open(input_file, 'r', encoding='utf-8') as file:
            total_lines = sum(1 for _ in file)

        # 每个部分的行数
        lines_per_part = total_lines // 4

        # 逐行读取并写入
        with open(input_file, 'r', encoding='utf-8') as file:
            current_part = 0
            current_line = 0
            output_file = open(output_files[current_part], 'w', encoding='utf-8')
            
            for line in file:
                output_file.write(line)
                current_line += 1
                
                # 检查是否需要切换到下一个文件
                if current_line >= lines_per_part and current_part < 3:
                    output_file.close()
                    current_part += 1
                    output_file = open(output_files[current_part], 'w', encoding='utf-8')
                    current_line = 0
            
            # 关闭最后一个文件
            output_file.close()

def print_file(infile, n=10):
    # 打开文件并读取前10行