Function write_csv_multithread

tools/io_file.py:509–529

使用多线程写入大CSV文件。 参数: - file_path (str): 文件路径。 - data (DataFrame): 要写入的CSV数据。 - mode (str): 写入模式，默认为 'a'（追加模式），可以选择 'w'（覆盖模式）。 - chunk_size (int): 数据大小阈值，单位为字节。 - encoding (str): 写入文件的编码格式，默认为 'utf-8'。

(file_path, data, mode, chunk_size, encoding='utf-8')

def write_csv_multithread(file_path, data, mode, chunk_size, encoding='utf-8'):
    """
    使用多线程写入大CSV文件。

    参数:
    - file_path (str): 文件路径。
    - data (DataFrame): 要写入的CSV数据。
    - mode (str): 写入模式，默认为 'a'（追加模式），可以选择 'w'（覆盖模式）。
    - chunk_size (int): 数据大小阈值，单位为字节。
    - encoding (str): 写入文件的编码格式，默认为 'utf-8'。
    """
    chunks = [data.iloc[i:i + chunk_size] for i in range(0, len(data), chunk_size)]
    cpu_count = os.cpu_count() // 2  # 限制使用一半的CPU核心数
    num_threads = min(cpu_count, len(chunks))

    logger.info(f"分割数据为 {len(chunks)} 个块进行多线程写入")

    with ThreadPoolExecutor(max_workers=num_threads) as executor:
        futures = [executor.submit(write_csv_chunk, file_path, chunk, mode, encoding) for chunk in chunks]
        for future in tqdm(futures, desc="写入CSV文件进度"):
            future.result()


def write_csv_chunk(file_path, chunk, mode, encoding='utf-8'):