pandas保存多个sheet|极客笔记

pandas保存多个sheet

在数据处理中，我们经常会遇到需要将数据保存到Excel文件的需求。而有时候，我们需要将数据保存到一个Excel文件中的多个sheet中。在Python中，可以使用pandas库来实现这一功能。

1. 创建多个DataFrame示例数据

首先，让我们创建一些示例数据，用于演示如何将多个DataFrame保存到一个Excel文件中的多个sheet中。

import pandas as pd

# 创建示例数据
data1 = {
    'Name': ['Alice', 'Bob', 'Charlie'],
    'Age': [25, 30, 35],
    'Gender': ['F', 'M', 'M']
}

data2 = {
    'City': ['New York', 'Los Angeles', 'Chicago'],
    'Income': [50000, 60000, 70000],
    'Occupation': ['Engineer', 'Teacher', 'Doctor']
}

df1 = pd.DataFrame(data1)
df2 = pd.DataFrame(data2)

2. 保存多个DataFrame到一个Excel文件的多个sheet

接下来，我们将使用pandas的ExcelWriter类将多个DataFrame保存到一个Excel文件的多个sheet中。

# 创建ExcelWriter对象
with pd.ExcelWriter('data.xlsx') as writer:
    df1.to_excel(writer, sheet_name='Sheet1', index=False)
    df2.to_excel(writer, sheet_name='Sheet2', index=False)

在上面的代码中，我们首先创建了一个ExcelWriter对象，并指定要保存的Excel文件名为data.xlsx。然后，使用to_excel方法将两个DataFrame分别保存到该Excel文件中的不同sheet中，分别命名为Sheet1和Sheet2。