利用python脚本实现身份证号文件到姓名文件的批量重命名
在文件管理中,经常需要根据特定条件修改文件名。本文介绍如何使用python脚本将以身份证号命名的文件重命名为对应的姓名。假设我们有一个excel文件,包含身份证号和姓名信息,我们将利用Python读取Excel数据并实现批量重命名。
原始代码存在效率问题,因为它在每次处理文件时都需要遍历整个Excel数据。改进后的代码通过直接遍历Excel数据,提高了效率。
原始代码:
import os import pandas as pd df = pd.read_excel("data.xls") id_list = os.listdir("./") for id_file in id_list: id_prefix = id_file[:id_file.find(".")] id_suffix = id_file[id_file.find("."):] name = df[df["身份证号"] == id_prefix].姓名.values if len(name) != 0: name = name[0] os.rename(id_file, name + id_suffix) print("重命名完成")
改进后的代码:
立即学习“Python免费学习笔记(深入)”;
import os import pandas as pd df = pd.read_excel("data.xls") id_name_map = {str(row[0]): row[1] for row in df.values} # 创建身份证号到姓名的映射字典 for id_file in os.listdir("./"): id_prefix = id_file[:id_file.find(".")] id_suffix = id_file[id_file.find("."):] if id_prefix in id_name_map: new_name = id_name_map[id_prefix] + id_suffix os.rename(id_file, new_name) print(f"已重命名: {id_file} -> {new_name}") print("重命名完成")
改进后的代码首先创建了一个身份证号到姓名的映射字典,这样可以O(1)的时间复杂度查找姓名,显著提高了效率。 它也更清晰地显示了重命名过程。 请确保“data.xls”文件存在于同一目录下,并且包含名为“身份证号”和“姓名”的列。 运行此脚本前,请备份您的文件,以防意外数据丢失。
© 版权声明
文章版权归作者所有,未经允许请勿转载。
THE END