将大型google sheets文件导出为批量大小为500的多个csv

import csv

batch_size = 500
file_prefix = "batch_"

def split_csv_into_batches(csv_file):
    with open(csv_file, 'r') as file:
        reader = csv.reader(file)
        header = next(reader)  # 读取文件头部

        batch_count = 1
        row_count = 0
        current_batch = []

        for row in reader:
            current_batch.append(row)
            row_count += 1

            if row_count == batch_size:
                write_batch_to_csv(current_batch, file_prefix + str(batch_count) + ".csv")
                batch_count += 1
                row_count = 0
                current_batch = []

        # 处理剩余的行数不足一个批次的情况
        if row_count > 0:
            write_batch_to_csv(current_batch, file_prefix + str(batch_count) + ".csv")

def write_batch_to_csv(batch, output_file):
    with open(output_file, 'w', newline='') as file:
        writer = csv.writer(file)
        writer.writerow(header)
        writer.writerows(batch)

# 使用示例
split_csv_into_batches("your_csv_file.csv")