python-封装方法用于读取excel

枫铃3年前 (2021-07-11)Python237

利用openpyxl方法实现读写excel表格

from openpyxl.reader.excel import load_workbook
import os
class ExcelMethod():
    def __init__(self, filename, sheetName):
        self.filename = filename
        self.wb = load_workbook(filename)
        # 通过工作的表名获取一个工作表对象
        self.sheet = self.wb[sheetName]
        # 获取工作表中的最大行号
        self.maxRowNum = self.sheet.max_row
        # 获取工作表中的最大列号
        self.max_column = self.sheet.max_column
        # 获取总的行
        self.row = self.sheet.max_row

    def readExcel(self):
        dataList = []
        try:
            for row in self.sheet.rows:
                tmpList = []
                for cell in row:
                    tmpList.append(cell.value)
                dataList.append(tmpList)
        except:
         print("%s加载失败" % self.filename)

        else:
         return dataList[1:]


    def saveExcel(self, row, text):
        try:
            self.sheet.cell(row, self.max_column, text)
            self.wb.save(self.filename)

        except:
            print("%s 保存失败" % self.filename)

if __name__=="__main__":
    dri_url = os.path.join(os.getcwd() + r"\User1.xlsx")
    excel = ExcelMethod(dri_url,"Sheet1")
    dataJson = excel.readExcel()
    print(dataJson)
    excel.saveExcel(4,'pass')

1.实现获取excel某张表的行数、单元格数据

'''
遇到问题没人解答?小编创建了一个Python学习交流QQ群:531509025
寻找有志同道合的小伙伴,互帮互助,群里还有不错的视频学习教程和PDF电子书!
'''
#coding=utf-8
import xlrd
#获取excel文件
data = xlrd.open_workbook('file_path/xxx.xls')#存放excle表格的路径
#获取第一张表数据
tables = data.sheets()[0]
#打印表行数
print(tables.nrows)
#打印第4行,第3列单元格数据
print(tables.cell_value(3,2))

2.封装获取表格方法

此方法需要实现的作用是:其他方法再调用此方法时,如果传入file_name和sheet_id,就调用对应路径的excel文件和对应的表。如果不传这两个字段,就调用默认表格。

def __init__(self,file_name=None,sheet_id=None):
        if file_name:
            self.file_name = file_name
            self.sheet_id = sheet_id
        else:
            self.file_name = 'file_path/xxx.xls'
            self.sheet_id = 0
        self.data = self.get_data()

3.封装获取表格数据方法

封装获取tables的方法,用以之后获取单元格行数、单元格数据,或其他表信息使用。

'''
遇到问题没人解答?小编创建了一个Python学习交流QQ群:531509025
寻找有志同道合的小伙伴,互帮互助,群里还有不错的视频学习教程和PDF电子书!
'''
def get_data(self):
        data = xlrd.open_workbook(self.file_name)
        tables = data.sheets()[self.sheet_id]
        return tables

4.封装获取单元格行数方法

def get_lines(self):
        tables = self.data
        return tables.nrows

5.封装获取单元格数据的方法

 def get_value(self,row,col):
        return self.data.cell_value(row,col)

6.封装获取总行数和总列数

'''
遇到问题没人解答?小编创建了一个Python学习交流QQ群:531509025
寻找有志同道合的小伙伴,互帮互助,群里还有不错的视频学习教程和PDF电子书!
'''
 def get_nrown_ncols(self):
         #获取总行数
         rowNum = self.data.nrows
        #获取总列数
         colNum = self.data.ncols
         return rowNum ,colNum 

7.写入数据到excel

def write_excel(self,row,column,value_back):
        '''
        :param row: 某一列
        :param value: 需要写入的值
        :return:
        '''
        wb=load_workbook(self.file_name)
        sheel = wb[self.sheel_name]
        #把值写到row,column组成的单元格
        sheel.cell(row,column).value = value_back
        #保存excel
        wb.save(self.file_name)

8.根据行号查找对应内容

'''
遇到问题没人解答?小编创建了一个Python学习交流QQ群:531509025
寻找有志同道合的小伙伴,互帮互助,群里还有不错的视频学习教程和PDF电子书!
'''
   #根据行号,找到该行的内容
   def get_row_values(self,row):
        tales = self.data
        row_data = tales.row_values(row)
        return row_data

整体代码如下:

#coding: utf-8
import xlrd

class OpeExcel:
    def __init__(self,file_name=None,sheet_id=None):
        if file_name:
            self.file_name = file_name
            self.sheet_id = sheet_id
        else:
            self.file_name = 'file_path/xxx.xls'
            self.sheet_id = 0
        self.data = self.get_data()

    #获取sheets的内容
    def get_data(self):
        data = xlrd.open_workbook(self.file_name)
        tables = data.sheets()[self.sheet_id]
        return tables

    #获取单元格行数
    def get_lines(self):
        tables = self.data
        return tables.nrows

    #获取单元格数据
    def get_value(self,row,col):
        return self.data.cell_value(row,col)

if __name__ == '__main__':
     opers = OpeExcel()
     print(opers.get_lines())
     print(opers.get_value(3,2))

9.给一个excel文件追加内容:

'''
遇到问题没人解答?小编创建了一个Python学习交流QQ群:531509025
寻找有志同道合的小伙伴,互帮互助,群里还有不错的视频学习教程和PDF电子书!
'''
from xlrd import open_workbook
from xlutils.copy import copy
import os
dri_url = os.path.join(os.getcwd()+r"\user.xls")
rexcel = open_workbook(dri_url)# 用wlrd提供的方法读取一个excel文件
rows = rexcel.sheets()[0].nrows  # 用wlrd提供的方法获得现在已有的行数
excel = copy(rexcel)  # 用xlutils提供的copy方法将xlrd的对象转化为xlwt的对象
table = excel.get_sheet(0)  # 用xlwt对象的方法获得要操作的sheet
table.write(1, 3, 'pass')  # xlwt对象的写方法,参数分别是行、列、值
excel.save(dri_url)  # xlwt对象的保存方法,这时便覆盖掉了原来的excel

python编辑已存在的excel坑: BadZipFile: File is not a zip file

为了能反复编辑已存在的excel文件并保存,需要xlwt、xlrd、xlutils组合起来使用,代码如下:

'''
遇到问题没人解答?小编创建了一个Python学习交流QQ群:531509025
寻找有志同道合的小伙伴,互帮互助,群里还有不错的视频学习教程和PDF电子书!
'''
import xlwt, os, xlrd
from xlutils.copy import copy


class Do_Excel:
      def __init__(self,filename,sheetname="Sheet1"):
          self.filename = filename
          self.sheetname = sheetname

      #读取excel
      def excel_read(self,x,y):
          data = xlrd.open_workbook(self.filename)
          tabel = data.sheet_by_name(self.sheetname)
          return tabel.cell_value(x,y)

      #判断exce文件是否存在,不存在则创建,存在则直接打开编辑
      def excel_create(self):
           if not os.path.exists(self.filename):
               data = xlwt.Workbook()
               table = data.add_sheet(self.sheetname)
               table.write(0,0,"id")
               data.save(self.filename)

       #综合xlw/xlrd/xlutis.copy 读写excle
      def write(self,row,col,value):
          self.excel_create()
          rb = xlrd.open_workbook(self.filename)
          wb = copy(rb)
          ws = wb.get_sheet(0)#1代表是写到第几个工作表里,从0开始算是第一个。
          ws.write(row,col,value)
          wb.save(self.filename)


if __name__=="__main__":
    dir = os.path.split(os.path.split(os.path.realpath(__file__))[0])[0]
    dir_excel  = os.path.join(dir+r"\data\user1.xlsx")
    Do_Excel(dir_excel).write(2,5,'pass')

解决办法二:直接使用openpyxl的Workbook和load_workbook,简单直接

rom openpyxl import Workbook,load_workbook
import os

class Do_Excel:
    def __init__(self,filename,sheetname='Sheet1'):
        self.filename=filename
        self.sheetname=sheetname

    def write(self,i,j,value):
        if not os.path.exists(self.filename):
            wb = Workbook()
            sh = wb.create_sheet(self.sheetname)
        else:
            wb = load_workbook(self.filename)
            sh = wb[self.sheetname]
        sh.cell(i,j).value=value
        wb.save(self.filename)

Do_Excel('test222.xlsx').write(1,1,'sdcds')
Do_Excel('test222.xlsx').write(1,2,'change')
Do_Excel('test222.xlsx').write(3,2,'pass')

相关文章

零基础学Python:函数的参数详解

必备参数关键字参数默认参...

python 中argparse 实例解析

一 概念: argparse是python的一个命令行解析包。它可以使写用户友好的命令行接口变得非常容易。该模块定义什么参数是需要的ÿ...

python列表list的基本性质

列表list是pytho...

Python3.8 新特性:赋值表达式

Python3.8 正式版本已经发布,喜欢尝鲜的同学可以下载下来体验一下,我并没有打算升级到最新版本,因为性能上并没...

Python高手必读,做一个精通规则的玩家

编程,其实和玩电子游戏有一些相似之处。你在玩不同游戏前,需要先学习每个游戏的不同规则,只有熟悉和灵活运用游戏规则&#...

发表评论

访客

看不清,换一张

◎欢迎参与讨论,请在这里发表您的看法和观点。