Python办公自动化：Pandas提取指定数据并保存在原Excel工作簿中

新建一个EXcel数据，然后写入数据，最后提取指定数据并保存在原Excel工作簿中

要使用 Pandas 提取指定数据并保存在原 Excel 工作簿中，您可以使用 pandas 和 openpyxl 库。

首先，安装这两个库：

pip install pandas
pip install openpyxl

该代码创建一个 Excel 文件，写入一些数据，然后从其中提取指定数据并保存在原 Excel 工作簿的新工作表中。

import pandas as pd

# 创建一个 Excel 文件，并写入数据
def create_excel_file(excel_file):
    df = pd.DataFrame({
        'Name': ['Alice', 'Bob', 'Charlie'],
        'Age': [25, 30, 35],
        'Occupation': ['Engineer', 'Doctor', 'Artist']
    })
    df.to_excel(excel_file, sheet_name='Sheet1', index=False)

# 从 Excel 文件中提取指定数据，并保存在原 Excel 工作簿中
def extract_and_save_to_same_workbook(excel_file):
    # 读取原始数据
    df = pd.read_excel(excel_file, sheet_name='Sheet1')
    
    # 提取指定数据（例如，年龄大于 30 的行）
    df_filtered = df[df['Age'] > 30]
    
    # 保存到原 Excel 工作簿的新工作表中
    with pd.ExcelWriter(excel_file, engine='openpyxl', mode='a') as writer:
        df_filtered.to_excel(writer, sheet_name='Filtered', index=False)

# 主程序
if __name__ == "__main__":
    excel_file = 'sample_data.xlsx'  # Excel 文件路径
    
    # 创建 Excel 文件，并写入数据
    create_excel_file(excel_file)
    
    # 提取指定数据，并保存在原 Excel 工作簿中 (xpanx.com)
    extract_and_save_to_same_workbook(excel_file)  # xpanx.com

    print("Data extraction and saving completed.")

知识点解释

1. 创建 Excel 文件和写入数据: 使用 pandas.DataFrame 创建一个数据框，并使用 to_excel 方法将其保存为 Excel 文件。
2. 读取 Excel 文件: 使用 pandas.read_excel 方法读取 Excel 文件。
3. 数据过滤: 使用 Pandas 的过滤功能来提取指定数据。
4. 保存到原 Excel 工作簿: 使用 pandas.ExcelWriter 类的 mode='a' 参数来追加数据到现有的 Excel 文件。

运行这个代码后，您将看到一个名为 sample_data.xlsx 的 Excel 文件，其中包含两个工作表：Sheet1（原始数据）和 Filtered（提取的数据）。