1,创建wb并测试
import logging
import os
import re
import pandas as pd
from openpyxl import Workbook, load_workbook
# 创建一个 Workbook 对象
wb = Workbook()
ws = wb.active
# 定义文件夹路径
folder_path = '.'
# 初始化一个空列表用于存储 Excel 文件路径
excel_files = []
# 使用 os.walk 函数递归遍历文件夹
for root, dirs, files in os.walk(folder_path):
# 遍历文件列表
for file in files:
# 检查文件名是否以".xlsx"或".xls"结尾,并且文件名中包含"机构信息"
if re.search(r'机构信息', file) and file.endswith('.xlsx'):
# 将文件的完整路径添加到列表中
excel_files.append(os.path.join(root, file))
# # 打印找到的 Excel 文件路径
# for file_path in excel_files:
# print(file_path)
# 遍历文件夹中的 Excel 文件
for i, file_name in enumerate(excel_files, start = 1):
if i == 1:
# 创建本excel文件中的新sheet页
# wb = load_workbook(os.path.join(folder_path, file_name))
# new_sheet = wb.create_sheet('新的 sheet 页')
# new_sheet['A1'] = "这是新工作表的内容"
# wb.save(os.path.join(folder_path, file_name))
with pd.ExcelFile(os.path.join(folder_path, file_name)) as xls:
# 遍历所有 sheet 页
for sheet_name in xls.sheet_names:
df = pd.read_excel(xls, sheet_name=sheet_name, usecols='A', header=0, engine='openpyxl', skiprows=0)
print(df)
2.自定义读到的excel并合并到新的excel文件中
import logging
import os
import re
import pandas as pd
from openpyxl import Workbook
# 创建一个 Workbook 对象
wb = Workbook()
ws = wb.active
# 定义文件夹路径
folder_path = '.'
# 初始化一个空列表用于存储 Excel 文件路径
excel_files = []
# 使用 os.walk 函数递归遍历文件夹
for root, dirs, files in os.walk(folder_path):
# 遍历文件列表
for file in files:
# 检查文件名是否以".xlsx"或".xls"结尾,并且文件名中包含"机构信息"
if re.search(r'机构信息', file) and file.endswith('.xlsx'):
# 将文件的完整路径添加到列表中
excel_files.append(os.path.join(root, file))
# # 打印找到的 Excel 文件路径
# for file_path in excel_files:
# print(file_path)
# 遍历文件夹中的 Excel 文件
for i, file_name in enumerate(excel_files, start = 1):
try:
# 读取当前 Excel 文件的数据
df = pd.read_excel(os.path.join(folder_path, file_name), usecols='A:D', header=0, engine='openpyxl',skiprows=0)
df = df.fillna('')
df.loc[-1] = file_name
# 增加一列用于存储文件名
df.loc[-2] = df.columns[0]
# df.columns = ['文件名'] + df.columns
# print(df)
# 初始化计数器
total_data = 0
for index, value in df.iterrows():
for col in df.columns:
if df.columns.get_loc(col) == 2:
total_data += 1
# print(total_data)
# print(df)
# print( col)
content_ws=ws.cell(row=i, column=total_data, value=df.values[index, df.columns.get_loc(col)]).value
# print(content_ws)
pass
except UserWarning as e:
# 记录异常信息
logging.basicConfig(filename='warning.log', level=logging.INFO)
# logger = logging.getLogger(__name__)
# logger.error(f"处理 {file_name} 时发生异常: {e}")
logging.info(f'Warning: {e}')
finally:
# # 记录处理完成的信息
print(f"合并了第{i}个文件: {file_name}")
wb.save('拆分后的机构.xlsx')