如何过滤DataDrame并生成多张表格的多个Excel文件

import pandas as pd

# 1. 过滤DataFrame
filtered_data = original_data[condition]  # 使用条件语句筛选出满足条件的数据

# 2. 分割DataFrame
subsets = []  # 存储划分后的子集
chunk_size = len(filtered_data) // num_files  # 计算每个子集的大小，num_files为需要生成的Excel文件数量
start = 0
for i in range(num_files):
    end = start + chunk_size
    subset = filtered_data.iloc[start:end]  # 划分子集
    subsets.append(subset)
    start = end
# 处理剩余的数据（如果数据量无法整除num_files）
if len(filtered_data) % num_files != 0:
    subset = filtered_data.iloc[start:]
    subsets.append(subset)

# 3. 生成Excel文件
for i, subset in enumerate(subsets):
    file_name = f"output_{i}.xlsx"  # Excel文件名，可以自定义命名规则
    sheet_name = "Sheet1"  # Excel表格名，可以自定义
    subset.to_excel(file_name, sheet_name=sheet_name, index=False)  # 保存为Excel文件，index参数指定是否保存索引