背景:接上一章,进一步增强功能,实现批量处理的功能。
import openpyxl
import os
# 定义一个字典
my_dict = {'dth1':'代替换1'
,'dth2':'代替换2'
,'dth3':'代替换3'
}
def change_column_name(path,my_dict):
wb = openpyxl.load_workbook(path)
ws = wb.active
for c in ws[1]:
if c.value.strip() in my_dict:
c.value=my_dict[c.value.strip()]
wb.save(path)
base_dir = r'E:\data1'
for fn in os.listdir(base_dir):
path=os.path.join(base_dir,fn)
print(path)
change_column_name(path,my_dict)