import os
import pandas as pd
import openpyxl
def add_title(data_path):
for dirpath,dirname,filenames in os.walk(data_path): #os.walk()遍历文件
for fname in filenames:
file_name=os.path.join(dirpath,fname) #os.path.join()合并路径
if file_name.endswith('.xlsx'):
wb = openpyxl.load_workbook(file_name)
ws = wb.worksheets[0] # 读sheet0
ws.insert_rows(0,1) # 在第0行前插入一行
ws.insert_cols(2,3) # 在第2列前插入三列
for index,row in enumerate(ws.rows):
if index == 0:
row[0].value = '客户号'
row[1].value = '证件类型'
row[2].value = '证件号'
row[3].value = '账号'
row[4].value = '日累计限额'
else:
row[1].value = None
row[2].value = None
row[3].value = None
wb.save(file_name)
print('finished...')
data_path= ""
add_title(data_path) #调用程序