pandas groupby to csv

如果你想把分组后的数据保存为CSV文件而不是Excel文件，你可以使用pandas的`to_csv`函数代替`to_excel`。以下是相应的修改版本： ```python import os import pandas as pd split_key = '外部ID' # 按照指定的key列进行分组 grouped = df_resultE.groupby(split_key) # 用于记录拆分后文件的序号 file_index = 1 # 用于记录每个文件已写入的行数 row_count = 0 # 每个文件的最大行数限制 max_rows_per_file = 25000 # 拆分后文件保存的目录，若不存在则创建 output_dir = 'split_files' if not os.path.exists(output_dir): os.makedirs(output_dir) csv_writer = None for key, group in grouped: # 如果当前文件的行数已经超过最大限制，则新建一个文件 if row_count >= max_rows_per_file: file_index += 1 row_count = 0 # 构造输出文件名 output_file = os.path.join(output_dir, f'split_{file_index}_{key}.csv') # 如果是新文件，关闭之前的CSV写入器（如果存在的话），然后初始化一个新的 if row_count == 0 and csv_writer is not None: csv_writer.close() csv_writer = open(output_file, 'w', newline='', encoding='utf-8') # 将分组数据写入CSV文件 group.to_csv(csv_writer, index=False) # 更新已写入的行数 row_count += len(group) # 最后别忘了关闭最后一个CSV文件 if csv_writer is not None: csv_writer.close() ``` 在这个版本中，我们使用了`open`函数创建一个CSV文件，然后在每次开始新文件前关闭之前打开的文件，并用`to_csv`将数据写入新的CSV。最后记得在循环结束后关闭最后一个文件。