python 使用openpyxl讀取excel數據
openpyxl介紹
openpyxl是一個開源項目,它是一個用於讀取/寫入Excel 2010文檔(如xlsx 、xlsm 、xltx 、xltm文件 )的Python庫,如果要處理更早格式的Excel文檔(xls),需要用到其它庫(如:xlrd、xlwt等),這是openpyxl比較其他模塊的不足之處。openpyxl是一款比較綜合的工具,不僅能夠同時讀取和修改Excel文檔,而且可以對Excel文件內單元格進行詳細設置,包括單元格樣式等內容,甚至還支持圖表插入、打印設置等內容。
python中與excel操作相關的模塊:
- 📔 xlrd庫:從excel中讀取數據,支持xls、xlsx格式
- 📔 xlwt庫:對excel進行修改操作,但不支持對xlsx格式的修改
- 📔 xlutils庫:在xlwt和xlrd中,對一個已存在文件進行修改
- 📘 openpyxl庫:主要針對xlsx格式的excel進行讀取和編輯
openpyxl安裝
pip install openpyxl
openpyxl基本概念
- WorkBook:工作簿對象
- Sheet:表單對象
- Cell:表格對象
也就是我們使用excel時可以看到的三個對象,我們操作excel的步驟是打開工作簿,選中表單,對單元格進行操作,而openpyxl的工作原理也是如此。
openpyxl對excel進行操作
新建excel
import openpyxl # 創建一個工作簿 workbook = openpyxl.Workbook() # 創建一個表單 sheet = workbook.create_sheet('表單1') # 寫入一個數據 sheet.cell(row=1, column=1, value="python") # 保存 workbook.save('test.xlsx')
🎈 註意點:如果保存的文件名已存在,將會覆蓋原來的文件
打開已存在的文件
- 打開已存在的文件:load_workbook()
- 獲取表單:workbook[‘表單名’]
- 讀取指定表格:cell(row, column).value
import openpyxl # 打開工作簿 workbook = openpyxl.load_workbook('test.xlsx') # 獲取表單 sheet = workbook['表單1'] # 讀取指定的單元格數據 cell = sheet.cell(row=1, column=1).value print(cell)
運行結果:
C:\software\python\python.exe D:/learn/python24/python_base/test.py python Process finished with exit code 0
讀取單個單元格
# 方式一:讀取A6單元格的值 cell1 = sheet['A6'].value # 方式二:讀取第3行,第4列單元格的值 cell2 = sheet.cell(row=3, column=4).value
讀取多個單元格
# 方式一:讀取A6單元格的值 cell1 = sheet['A6'].value # 方式二:讀取第3行,第4列單元格的值 cell2 = sheet.cell(row=3, column=4).value
讀取多個單元格時,都是以元組的形式返回
其他相關操作
關鍵代碼 | 相關描述 |
---|---|
cell(row, column, value) | 寫入數據,三個參數分別是行,列,值 |
cell(row, column).value | 獲取指定單元格的值,兩個參數分別是行,列 |
workbook.remove(‘表單名’) | 刪除表單 |
del workbook[‘表單名’] | 刪除表單 |
workbook.save(‘文件名’) | 保存工作簿 |
sheet.max_row | 獲取表單數據的總行數 |
sheet.max_column | 獲取表單數據的總列數 |
sheet.rows | 獲取按行所有的數據 |
sheet.columns | 獲取按列所有的數據 |
🎈 註意點:
- openpyxl中列和行的起始標識都是1,不是從0開始
- 從excel中讀取出來的數據隻有兩種類型,即數值類型和字符串類型
- 不要隨便在表格中敲空格,會影響判斷最大行數和最大列數
- 運行操作excel的代碼時,要先關閉在操作系統中打開的相關excel表,否則可能會無法讀取/寫入數據
- 寫入數據時要執行保存
import openpyxl # 打開工作簿 wb = openpyxl.load_workbook('test.xlsx') # 獲取表單 sh = wb['表單1'] # 讀取指定的單元格數據 res1 = sh.cell(row=1, column=1).value print(res1) # 獲取最大行數 print(sh.max_row) # 獲取最大列數 print(sh.max_column) # 按列讀取所有數據,每一列的單元格放入一個元組中 print(sh.columns) # 直接打印,打印結果是一個可迭代對象,我們可以轉換成列表來查看 # 按行讀取所有數據,每一行的單元格放入一個元組中 rows = sh.rows # print(list(rows)) # 轉換成列表之後打印結果為具體的單元格,如下 # [(<Cell '表單1'.A1>, <Cell '表單1'.B1>, <Cell '表單1'.C1>, <Cell '表單1'.D1>, <Cell '表單1'.E1>, <Cell '表單1'.F1>, <Cell '表單1'.G1>), # (<Cell '表單1'.A2>, <Cell '表單1'.B2>, <Cell '表單1'.C2>, <Cell '表單1'.D2>, <Cell '表單1'.E2>, <Cell '表單1'.F2>, <Cell '表單1'.G2>), # (<Cell '表單1'.A3>, <Cell '表單1'.B3>, <Cell '表單1'.C3>, <Cell '表單1'.D3>, <Cell '表單1'.E3>, <Cell '表單1'.F3>, <Cell '表單1'.G3>)] # 我們可以通過for循環以及value來查看單元格的值 for row in list(rows): # 遍歷每行數據 case = [] # 用於存放一行數據 for c in row: # 把每行的每個單元格的值取出來,存放到case裡 case.append(c.value) print(case)
運行結果:
C:\software\python\python.exe D:/learn/python24/python_base/test.py case_id 3 7 <generator object Worksheet._cells_by_col at 0x0000012FEF90D8B8> ['case_id', 'interface', 'title', 'method', 'url', 'data', 'expected'] [1, 'login', '登錄成功', 'post', '/member/login', '{"mobile_phone":"15692004245","pwd":"miki12345"}', "{'code': 0, 'msg': 'OK'}"] [2, 'login', '登錄失敗-手機號為空', 'post', '/member/login', '{"mobile_phone":"","pwd":"miki12345"}', "{'code': 2, 'msg': '手機號碼為空'}"]
excel表中的原數據如下圖:
openpyxl封裝
在測試過程中,我們會經常需要訪問excel讀取測試數據,如果毎訪問一次我們就調用一次openpyxl,進行打開excel,選取表單,讀取數據等操作,將會增加我們的工作量和代碼量。凡是需要重復使用的,我們都可以對它進行二次封裝,寫成我們自己的版本,還可以在封裝過程中加入我們想要的內容,使用的時候直接調用即可。
在封裝openpyxl時,讀取excel時我寫瞭兩種實現方法,第一種是讀取數據後,把每條測試數據的鍵值保存到一個字典中,再把字典存到一個列表,這種方式比較容易理解,但是調用的時候需要多敲幾個括號或者引號;第二種是把每條測試數據的鍵值存到一個實例對象的屬性中,再把每個實例存到一個列表。實際上他們的作用是一樣,隻是第二種方式調用時更簡單。
# read_excel.py import openpyxl class CasesData: """用於保存測試用例數據""" pass class ReadExcel: def __init__(self, file_name, sheet_name): self.file_name = file_name self.sheet_name = sheet_name def open(self): self.wb = openpyxl.load_workbook(self.file_name) self.sh = self.wb[self.sheet_name] def close(self): self.wb.close() def read_data(self): """按行讀取數據,最後返回一個存儲字典的列表""" self.open() rows = list(self.sh.rows) titles = [] for t in rows[0]: title = t.value titles.append(title) cases = [] for row in rows[1:]: case = [] for r in row: case.append(r.value) cases.append(dict(zip(titles, case))) # 通過zip聚合打包用例的標題和數據 self.close() return cases def read_data_obj(self): """按行讀取數據,最後返回的是一個存儲實例對象的列表""" self.open() rows = list(self.sh.rows) titles = [] for t in rows[0]: title = t.value titles.append(title) cases = [] for row in rows[1:]: case_obj = CasesData() # 創建一個對象用來保存該行用例數據 case = [] for r in row: case.append(r.value) case_info = dict(zip(titles, case)) for k, v in case_info.items(): setattr(case_obj, k, v) cases.append(case_obj) self.close() return cases def write_data(self, row, column, msg): self.open() self.sh.cell(row=row, column=column, value=msg) self.wb.save(self.file_name) self.close() if __name__ == '__main__': # 直接運行本文件時執行,下面是一個應用實例 # 需要讀取excel時直接調用ReadExcel類 test = ReadExcel('cases.xlsx', 'login') res = test.read_data_obj() # 最後返回的是一個存儲實例對象的列表 print(res[0].case_id) # 打印第1個用例的case_id print(res[3].title) # 打印第4個用例的title test2 = ReadExcel('cases.xlsx', 'login') res2 = test2.read_data() # 最後返回一個存儲字典的列表 print(res2[0]['case_id']) # 打印第1個用例的case_id print(res2[3]['title']) # 打印第4個用例的title
上面封裝的openpyxl中,應用實例的excel表結構如下:
以上就是python 使用openpyxl讀取excel數據的詳細內容,更多關於python openpyxl讀取excel數據的資料請關註WalkonNet其它相關文章!
推薦閱讀:
- 詳解Python操作Excel之openpyxl
- Python 操作 Excel 之 openpyxl 模塊
- python技巧分享Excel創建和修改
- python使用openpyxl庫讀寫Excel表格的方法(增刪改查操作)
- python之openpyxl模塊的安裝和基本用法(excel管理)