使用Python封装excel操作指南

前言

openpyxl 是 python 中操作 excel 表格比较常用的一个库,可以读取和写入excel文件,支持【.xlsx / .xlsm / .xltx / .xltm】格式的文件,处理excel数据、公式、样式,且可以在表格内插入图表

但是在实际项目的使用过程中,如果经常要用到 openpyxl 进行操作,进行相应的封装,会事半功倍

结构说明

- - config   配置文件夹
- - - - dir_config.py  配置文件路径等相关信息
- - - - excel_handler.py  用于封装excel操作
- - exceldir 表格文件存放路径
- - run.py   执行文件

代码解析

excel_handler.py

from openpyxl import load_workbookfrom openpyxl.worksheet.worksheet import Worksheetclass ExcelHandler(): ''' 操作Excel ''' def __init__(self, file):  '''初始化函数'''  self.file = file def open_sheet(self, sheet_name) -> Worksheet:  '''打开表单'''  wb = load_workbook(self.file)  sheet = wb[sheet_name]  return sheet def read_header(self, sheet_name):  '''获取表单的表头'''  sheet = self.open_sheet(sheet_name)  headers = []  for i in sheet[1]:   headers.append(i.value)  return headers def read_rows(self,sheet_name):  '''  读取除表头外所有数据(除第一行外的所有数据)  返回的内容是一个二维列表,若想获取每一行的数据,可使用for循环或*解包  '''  sheet = self.open_sheet(sheet_name)  rows = list(sheet.rows)[1:]  data = []  for row in rows:   row_data = []   for cell in row:    row_data.append(cell.value)   data.append(row_data)  return data def read_key_value(self,sheet_name):  '''  获取所有数据,且将表头中的内容与数据结合展示(以字典的形式)  如:[  {'序号':1,'会员卡号': '680021685898','机场名称':'上海机场'},  {'序号':2,'会员卡号': '680021685899','机场名称':'广州机场'}  ]  '''  sheet = self.open_sheet(sheet_name)  rows = list(sheet.rows)  # 获取标题  data = []  for row in rows[1:]:   rwo_data = []   for cell in row:    rwo_data.append(cell.value)    # 列表转换成字典,与表头里的内容使用zip函数进行打包   data_dict = dict(zip(self.read_header(sheet_name),rwo_data))   data.append(data_dict)  return data @staticmethod def write_change(file,sheet_name,row,column,data):  '''写入Excel数据'''  wb = load_workbook(file)  sheet = wb[sheet_name]  # 修改单元格  sheet.cell(row,column).value = data  # 保存  wb.save(file)  # 关闭  wb.close()

写入Excel数据这一步,使用了静态方法,原因是读取文件可以无需保存,而修改文件后,如果没有进行保存,而其他地方有调用了该方法,则会引起一些报错,所以,每对excel进行一次修改,都进行一次保存

dir_config.py

import os# ExcelDemo 所在的绝对路径base_dir = os.path.split(os.path.split(os.path.abspath(__file__))[0])[0]# excel 文件放置目录excel_dir = os.path.join(base_dir,'excedir')

run.py

import osfrom ExcelDemo.config.excel_handler import ExcelHandlerfrom ExcelDemo.config import dir_configexcelfile = os.path.join(dir_config.excel_dir,'航班信息.xlsx')if __name__ == '__main__': excel = ExcelHandler(excelfile) header = excel.read_header('Sheet1') data = excel.read_rows('Sheet1') data2 = excel.read_key_value('Sheet1') print(header) ''' ['序号', '会员卡号', '机场', '航班日期', '航班号'] ''' print('**********************************') print(data) ''' [ [1, '680021685898', '西安南航', 20200503, 'CZ6754'], [2, '189000177074', '(新疆)莎车机场', 20200603, 'CZ6880'],  [3, '480005387697', '新疆南航', 20200612, 'CZ5390'],  [4, '380025990156', '西安南航', 20200619, 'CZ6622'] ]  ''' print('**********************************') print(data2) ''' [ {'序号': 1, '会员卡号': '680021685898', '机场': '西安南航', '航班日期':20200503, '航班号': 'CZ6754'}, {'序号': 2, '会员卡号': '189000177074', '机场': '(新疆)莎车机场', '航班日期': 20200603, '航班号': 'CZ6880'},  {'序号': 3, '会员卡号': '480005387697', '机场': '(新疆)库车', '航班日期': 20200612, '航班号': 'CZ5390'}, {'序号': 4, '会员卡号': '380025990156', '机场': '西安南航', '航班日期': 20200619, '航班号': 'CZ6622'} ]  ''' excel.write_change(excelfile,'Sheet1',6,5,'新增内容') excel.write_change(excelfile,'Sheet1',2,3,'修改内容')

修改后表格内容

总结

到此这篇关于使用python封装excel操作的文章就介绍到这了,更多相关python封装excel操作内容请搜索 以前的文章或继续浏览下面的相关文章希望大家以后多多支持 !

相关文章

发表新评论