Python封装excel操作

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_workbook
from openpyxl.worksheet.worksheet import Worksheet


class 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 os

from ExcelDemo.config.excel_handler import ExcelHandler
from ExcelDemo.config import dir_config

excelfile = 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,'修改内容')

修改后表格内容
在这里插入图片描述

  • 4
    点赞
  • 24
    收藏
    觉得还不错? 一键收藏
  • 4
    评论
Python中有多种库可以用来封装Excel,其中比较流行的有openpyxl和xlwt/xlrd。下面分别介绍这两种库的用法: 1. openpyxl openpyxl是一个非常强大的用于读写Excel文件的Python库,它可以处理Excel 2010及以上版本的.xlsx文件。下面是一个使用openpyxl库封装Excel的示例代码: ```python # 导入openpyxl库 from openpyxl import Workbook, load_workbook # 创建一个新的Excel文件 wb = Workbook() # 获取当前活动的工作表 ws = wb.active # 在A1单元格写入数据 ws['A1'] = 'Hello, world!' # 保存文件 wb.save('example.xlsx') # 打开一个已有的Excel文件 wb = load_workbook(filename='example.xlsx') # 获取当前活动的工作表 ws = wb.active # 读取A1单元格的数据 print(ws['A1'].value) ``` 2. xlwt/xlrd xlwt是Python中用于写Excel文件的库,xlrd则是用于读Excel文件的库。它们都支持.xls格式的文件。下面是一个使用xlwt/xlrd库封装Excel的示例代码: ```python # 导入xlwt和xlrd库 import xlwt import xlrd # 创建一个新的Excel文件 wb = xlwt.Workbook(encoding='utf-8') # 添加一个工作表 ws = wb.add_sheet('Sheet1') # 在A1单元格写入数据 ws.write(0, 0, 'Hello, world!') # 保存文件 wb.save('example.xls') # 打开一个已有的Excel文件 wb = xlrd.open_workbook('example.xls') # 获取第一个工作表 ws = wb.sheet_by_index(0) # 读取A1单元格的数据 print(ws.cell_value(0, 0)) ``` 以上就是使用Python封装Excel的方法,具体选择哪个库取决于你的需求和Excel文件的版本。
评论 4
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值