Python自動化如何處理excel文件數據

在python自動化中，經常會遇到對數據文件的操作，比如添加多名員工，但是直接將員工數據寫在python文件中，不但工作量大，要是以後再次遇到類似批量數據操作還會寫在python文件中嗎？

應對這一問題，可以將數據寫excel文件，針對excel 文件進行操作，完美解決。

本文僅介紹python對excel的操作。

安裝xlrd 庫

xlrd庫 官方地址：https://pypi.org/project/xlrd/

pip install xlrd

筆者在安裝時使用了 pip3 install xlrd

原因：筆者同時安裝了python2 和 python3

如果pip的話會默認將庫安裝到python2中，python3中不能直接調用。

那麼到底是使用pip 還是pip3進行安裝呢？

如果系統中只安裝了Python2，那麼就只能使用pip。

如果系統中只安裝了Python3，那麼既可以使用pip也可以使用pip3，二者是等價的。

如果系統中同時安裝了Python2和Python3，則pip默認給Python2用，pip3指定給Python3用。

Xlrd 庫簡單的使用

以如下excel文件為例進行操作

文件名為demo，有兩個sheet，名為工作表1和工作表2

工作表1中有如下數據

簡單的使用

# coding=utf-8

import xlrd

# 打開文件
data = xlrd.open_workbook('file/demo.xlsx')

# 查看工作表
data.sheet_names()
print("sheets：" + str(data.sheet_names()))

# 通過文件名獲得工作表,獲取工作表1
table = data.sheet_by_name('工作表1')

# 列印data.sheet_names()可發現，返回的值為一個列表，通過對列表索引操作獲得工作表1
# table = data.sheet_by_index(0)

# 獲取行數和列數
# 行數：table.nrows
# 列數：table.ncols
print("總行數：" + str(table.nrows))
print("總列數：" + str(table.ncols))

# 獲取整行的值 和整列的值，返回的結果為數組
# 整行值：table.row_values(start,end)
# 整列值：table.col_values(start,end)
# 參數 start 為從第幾個開始列印，
# end為列印到那個位置結束，默認為none
print("整行值：" + str(table.row_values(0)))
print("整列值：" + str(table.col_values(1)))

# 獲取某個單元格的值，例如獲取B3單元格值
cel_B3 = table.cell(3,2).value
print("第三行第二列的值：" + cel_B3)

運行後結果

項目中使用

獲得所有的數據

# coding=utf-8

import xlrd

def read_xlrd(excelFile):
   data = xlrd.open_workbook(excelFile)
   table = data.sheet_by_index(0)

   for rowNum in range(table.nrows):
       rowVale = table.row_values(rowNum)
       for colNum in range(table.ncols):
           if rowNum > 0 and colNum == 0:
               print(int(rowVale[0]))
           else:
               print(rowVale[colNum])
       print("")

   # if判斷是將 id 進行格式化
   # print("未格式化Id的數據：")
   # print(table.cell(1, 0))
   # 結果：number:1001.0


if __name__ == '__main__':
   excelFile = 'file/demo.xlsx'
   read_xlrd(excelFile=excelFile)

結果

如果在項目中使用則可將內容方法稍為做修改，獲得所有的數據後，將每一行數據作為數組進行返回

# coding=utf-8

import xlrd

def read_xlrd(excelFile):
   data = xlrd.open_workbook(excelFile)
   table = data.sheet_by_index(0)
   dataFile = []

   for rowNum in range(table.nrows):
       # if 去掉表頭
       if rowNum > 0:
           dataFile.append(table.row_values(rowNum))

   return dataFile


if __name__ == '__main__':
   excelFile = 'file/demo.xlsx'
   print(read_xlrd(excelFile=excelFile))

結果

testfan一直致力於提高整個測試行業技術實力，近期特推出」大咖解題--公益提升計劃「項目，幫助所有測試從業者走上高端測試之路！

點擊文章底部的閱讀原文：

可報名參加本期解題--7月28日晚上9點「面試常問的幾個Linux命令你真的會用嗎？」

掃上圖加安然老師微信可解鎖上期解題視頻，暗號：「規劃接口平臺需要哪些技術」