使用Python自动化更新Excel数据:从映射文件到批量替换完整指南

作品介绍

作品名称：从映射文件到批量替换的完整指南

用到的库：os、pandas

作品简介：该实例用于自动化更新Excel文件中的数据。它通过读取一个映射文件（替换格式.xlsx），然后根据映射关系批量替换目标目录中所有Excel文件中的数据。

实现过程

一、导入必要的库

import os
import pandas as pd

二、加载映射关系

1. 使用pandas.read_excel读取Excel文件。
2. 假设映射文件的第一列是“键”，第二列是“值”，通过zip函数将它们组合成一个字典。
3. 如果读取过程中发生错误（例如文件不存在或格式错误），捕获异常并返回一个空字典。

def load_mapping(file_path):
    """加载替换格式.xlsx中的映射关系"""
    try:
        df = pd.read_excel(file_path)
        mapping = dict(zip(df.iloc[:, 0], df.iloc[:, 1]))
        return mapping
    except Exception as e:
        print(f"加载映射文件'{file_path}'时出错: {e}")
        return {}

三、更新Excel文件

1. 使用os.listdir获取目录中的所有文件。
2. 检查文件扩展名是否为.xlsx或.xls，以确定是否是Excel文件。
3. 根据文件扩展名选择合适的engine（openpyxl用于.xlsx文件，xlrd用于.xls文件）。
4. 读取Excel文件并逐行检查：
5. 如果第一列的值在映射字典中存在，则将其替换为映射字典中的对应值。
6. 保存修改后的Excel文件。

def update_excel_files(directory, mapping):
    """遍历目录中的所有Excel文件并更新数据"""
    for filename in os.listdir(directory):
        if filename.endswith('.xlsx') or filename.endswith('.xls'):
            file_path = os.path.join(directory, filename)


            try:
                # 根据文件扩展名选择合适的engine
                engine = 'openpyxl' if filename.endswith('.xlsx') else 'xlrd'


                # 读取Excel文件
                df = pd.read_excel(file_path, engine=engine)


                # 替换数据
                if len(df.columns) > 1:  # 确保至少有两列
                    for index, row in df.iterrows():
                        if row[0] in mapping:
                            df.at[index, df.columns[0]] = mapping[row[0]]


                # 保存修改后的Excel文件
                df.to_excel(file_path, index=False, engine=engine)
                print(f"成功处理文件: {filename}")
            except Exception as e:
                print(f"处理文件'{file_path}'时出错: {e}")

四、主程序逻辑

1. 指定映射文件路径（替换格式.xlsx）并调用load_mapping函数加载映射关系。
2. 如果映射关系加载成功（即mapping不为空），指定目标目录（./产品）并调用update_excel_files函数更新目录中的所有Excel文件。
3. 如果映射关系加载失败，输出错误信息并终止程序。

# 加载映射关系
mapping_file = './替换格式.xlsx'
mapping = load_mapping(mapping_file)
print("提取的映射关系:", mapping)


if mapping:  # 只有当成功加载映射关系时才继续
    # 目标目录
    directory = './产品'


    # 更新目录下的所有Excel文件
    update_excel_files(directory, mapping)


    print("所有Excel文件的数据已成功替换并保存。")
else:
    print("由于映射关系加载失败，无法进行数据替换。")

五、结语

以上就是本次分享的全部内容，如果你有任何疑问或想要分享的经验，评论区永远为你敞开。你的每一个点赞和转发，都是对我们最大的支持和鼓励！