使用python根据文件夹中的长度拆分多个csv文件

import os
import csv

def split_csv_files(folder_path, threshold):
    # 获取文件夹中的所有文件名
    file_names = os.listdir(folder_path)
    
    for file_name in file_names:
        file_path = os.path.join(folder_path, file_name)
        
        # 获取文件的长度
        file_size = os.path.getsize(file_path)
        
        if file_size > threshold:
            # 拆分文件
            with open(file_path, 'r') as file:
                reader = csv.reader(file)
                header = next(reader)  # 读取CSV文件的头部
                
                # 创建新的CSV文件
                count = 1
                for row in reader:
                    if count % threshold == 1:
                        # 每个阈值创建一个新的CSV文件
                        new_file_path = os.path.join(folder_path, f'{file_name}_{count}.csv')
                        with open(new_file_path, 'w', newline='') as new_file:
                            writer = csv.writer(new_file)
                            writer.writerow(header)  # 写入头部
                            writer.writerow(row)  # 写入数据
                    else:
                        # 继续写入当前CSV文件
                        with open(new_file_path, 'a', newline='') as new_file:
                            writer = csv.writer(new_file)
                            writer.writerow(row)  # 写入数据
                    
                    count += 1
        else:
            # 文件长度不超过阈值，不需要拆分
            continue