要实现这个目标,可以遍历数据框并将每个工作表添加到新的xlsx文件中。
以下是示例代码:
import pandas as pd
# 创建示例数据框
data = {'Name': ['Tom', 'Jerry', 'Spike', 'Tyke'],
'Age': [5, 3, 7, 1],
'Breed': ['Cat', 'Mouse', 'Dog', 'Dog']}
df = pd.DataFrame(data)
# 创建新的Excel工作簿
writer = pd.ExcelWriter('data.xlsx', engine='xlsxwriter')
# 遍历数据框中的每个工作表,并将其保存在新的工作表中
for key, item in df.groupby('Breed'):
item.to_excel(writer, sheet_name=key, index=False)
# 保存并关闭ExcelWriter对象
writer.save()
在这个示例中,我们使用groupby()函数按照Breed列对数据框进行分组。遍历每个分组,将其保存到单独的工作表中,每个工作表都被命名为分组的键。最后,ExcelWriter对象被保存并关闭以保存新的xlsx文件。