Python使用openpyxl操作excel表格

前言

本文的文字及圖片來源於網絡,僅供學習、交流使用,不具有任何商業用途,版權歸原作者所有,如有問題請及時聯系我們以作處理。

PS：如有需要Python學習資料的小伙伴可以加點擊下方鏈接自行獲取

http://note.youdao.com/noteshare?id=3054cce4add8a909e784ad934f956cef

一、openpyxl的使用

pip install openpyxl

第一步：打開工作簿（讀取Excel文件中所有的數據保存為工作簿對象）

workbook = openpyxl.load_workbook('cases.xlsx')

第二步：選中表單對象

sheet = workbook['login']

第三步：通過表單選中表格讀取數據

①讀取內容（第五行第四列）

data = sheet.cell(row=5,column=4)
printa(data.value) # 獲取內容用value方法

②寫入內容（保存才會生效）

sheet.cell(row=7,column=3,value='("Python","123456")')
# 寫入內容后，一定要保存才會生效
workbook.save('cases.xlsx')

③獲取最大行和最大列

# 最大行
sheet.max_row
# 最大列
sheet.max_column

④按行獲取所有的格子對象，每一行格子放入一個元組中

sheet.rows

注意點：不要隨便在表格中敲空格

二、openpyxl的封裝

導包

import openpyxl

 

封裝

class ReadExcel(object):
    """操作Excl文件"""

    def __init__(self, fileName, sheetName):
        """
        初始化方法
        :param fileName: Excel文件名
        :param sheetName: 表單名
        """
        self.fileName = fileName
        self.sheetName = sheetName

    def open(self):
        """打開工作簿，選中表單"""
        self.wb = openpyxl.load_workbook(self.fileName)
        self.sh = self.wb[self.sheetName]

    def save(self):
        """保存工作簿對象的方法"""
        self.wb.save(self.fileName)
        self.wb.close()  # 這一行加不加關系不大，加了可以釋放內存

 

1、讀取數據

①將每條數據存儲成字典類型

    def read_data_dict(self):
        """讀取數據(將每條數據存儲成字典類型)"""
        # 打開工作簿
        self.open()
        # 通過rows獲取Excel文件中所有的行數據，然后把數據轉成列表
        rows = list(self.sh.rows)
        # 表頭
        title = []
        # 遍歷Excel文件中的第一行表頭信息
        for i in rows[0]:
            title.append(i.value)
        # 用例數據列表
        cases = []
        # 遍歷用例數據行
        for row in rows[1:]:
            # 定義一個列表存放每一行的數據
            data = []
            # 遍歷每一行的數據
            for r in row:
                data.append(r.value)
            # 把每一行的數據通過zip進行打包，然后轉成字典，存入到用例數據列表中
            case = dict(zip(title, data))
            cases.append(case)
        return cases

 

②將每條數據存儲成CaseData類對象類型

class CaseData(object):
    """保存用例數據類"""
    pass
    def read_data_obj(self):
        """讀取數據(將數據保存到CaseData類中)"""
        self.open()
        rows = list(self.sh.rows)
        # 表頭
        title = []
        # 遍歷Excel文件中的第一行表頭信息
        for i in rows[0]:
            title.append(i.value)
        # 定義用例列表，用來存放用例類列表
        cases = []
        # 遍歷用例數據行
        for row in rows[1:]:
            # 定義一個列表存放每一行的數據
            data = []
            # 遍歷每一行的數據
            for r in row:
                data.append(r.value)
            # 把每一行的數據通過zip進行打包，然后轉成字典，存入到用例數據列表中
            case = dict(zip(title, data))
            # 定義一個用例存放類對象
            case_obj = CaseData()
            for k, v in case.items():
                # 通過setattr()給對象添加屬性
                setattr(case_obj, k, v)
            # 把對象添加到列表中
            cases.append(case_obj)
        return cases

 

2、寫入數據

    def write_data(self, row, column, value):
        """
        寫入數據
        :param row: 行
        :param column: 列
        :param value: 數據
        """
        self.open()
        # 指定行列進行寫入數據
        self.sh.cell(row=row, column=column, value=value)
        # 保存
        self.wb.save(self.fileName)
        # 關閉
        self.wb.close()