python 读取excel封装实例

最新推荐文章于 2021-11-12 16:27:19 发布

似水@流年

最新推荐文章于 2021-11-12 16:27:19 发布

阅读量701

点赞数 1

分类专栏： Python 其他

本文链接：https://blog.csdn.net/mr_hui_/article/details/104506622

版权

Python 同时被 2 个专栏收录

16 篇文章 1 订阅

订阅专栏

其他

1 篇文章 0 订阅

订阅专栏


import xlrd
from xlutils.copy import copy


class OperateExcel(object):
    def __init__(self, file_name=None, sheet_id=None):
        """
        :param file_name: excel文件的具体路径名称
        :param sheet_id:  要操作的第几 sheet 页
        """
        if file_name:
            self.file_name = file_name
        else:
            # self.file_name = '../data/util_data/operate_excel.xls'
            # self.file_name = os.path.join(basedir, "../data/case.xls")
            self.file_name = r'D:\project\WebAutoTest\Interface_demo\data\case.xls'
            print(self.file_name)
          

        if sheet_id:
            self.sheet_id = sheet_id
        else:
            self.sheet_id = 0

        self.sheet_table = self.get_sheet()

    # 获取 sheet 页操作对象
    def get_sheet(self):
        data = xlrd.open_workbook(self.file_name)
        sheet_table = data.sheets()[self.sheet_id]
        return sheet_table

    # 获取该 sheet 页的行数和列数，拿到的是一个元组
    def get_sheet_nrows_ncols(self):
        return self.sheet_table.nrows, self.sheet_table.ncols

    # 获取该 sheet 页的行数
    def get_sheet_nrows(self):
        return self.sheet_table.nrows

    # 获取该 sheet 页的列数
    def get_sheet_ncols(self):
        return self.sheet_table.ncols

    # 获取具体单元格的数据
    def get_sheet_cell(self, row, col):
        """
        :param row: 单元格的行值
        :param col: 单元格的列值
        :return: cell_data
        """
        cell_data = self.sheet_table.cell_value(row, col)
        return cell_data

    # 写入数据到 excel 中
    def write_to_excel(self, row, col, value):

        # 同样的先打开 excel 操作句柄
        data = xlrd.open_workbook(self.file_name)
        copy_data = copy(data)
        # 选择写入的 sheet 页
        copy_data_sheet = copy_data.get_sheet(0)
        # 写入数据
        copy_data_sheet.write(row, col, value)
        # 保存数据
        copy_data.save(self.file_name)


if __name__ == "__main__":
    oe = OperateExcel()
    print(oe.get_sheet_nrows_ncols())
    print(oe.get_sheet_nrows())
    print(oe.get_sheet_ncols())
    print(oe.get_sheet_cell(1, 1))
    print(oe.get_sheet_cell(1, 2))
    print(oe.get_sheet_cell(2, 2))
    oe.write_to_excel(17, 7, 'hello world')

注意,这边用xlrd读取的文件不能直接写,会报错显示没有write函数,避免这个坑就是用xlutils.copy一份出来