如何在python中使用openpyxl讀取excel數據

如何在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

讀取多個單元格時，都是以元組的形式返回

其他相關操作

注意點：

• 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

關于如何在python中使用openpyxl讀取excel數據問題的解答就分享到這里了，希望以上內容可以對大家有一定的幫助，如果你還有很多疑惑沒有解開，可以關注億速云行業資訊頻道了解更多相關知識。