利用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”文件存在于同一目录下，并且包含名为“身份证号”和“姓名”的列。 运行此脚本前，请备份您的文件，以防意外数据丢失。
