xlrd实现从excel文件读取数据

自动化测试时，经常会提到数据驱动的概念。数据驱动是以数据来驱动整个测试用例的执行，即测试数据决定测试结果。Python中ddt模块可以实现。python操作excel主要用到xlrd和xlwt这两个库，即xlrd是读excel，xlwt是写excel的库。

如果测试数据写入到excel文件中，该如何从中获取数据呢?

● 假设测试登录功能的用例：

● 正常账号密码登录成功;

● 错误账号登录失败;

● 错误密码登录失败;

● ......

根据当前系统数据，准备好三组测试用例的数据，写入到excel表格中：

读取数据使用xlrd(支持xls格式)示例：

# xlrd是第三方模块需要安装： pip install xlrd

import xlrd

def get_data_from_excel(filename,sheetname):

# 创建工作簿对象

workbook = xlrd.open_workbook(filename)

# 获取sheet对象

sheet = workbook.sheet_by_name(sheetname)

# 存放读取到数据，列表

datas = []

# 获取每列的列名存放到到keys

keys = [sheet.cell_value(0, col) for col in range(sheet.ncols)]

# 读取数据

# 有几行则循环几次，并忽略第一行表头的读取

for row in range(1,sheet.nrows):

# 存放一行中读取到的所有内容，字典格式

line_data = {}

# 循环遍历一行中的所有单元格

for col in range(sheet.ncols):

# 获取到单元格内容后，在keys中找到其列名，存放到字典中

line_data[keys[col]] = sheet.cell_value(row, col)

# 把读取到一行内容添加到datas

datas.append(line_data)

# 返回读取到的数据

return datas

附Xlrd常用操作如下：

打开excel:

path = r"d:\xlrd_test.xlsx"

wb = xlrd.open_workbook(path, encoding_override="utf-8")

table = wb.sheets()[0] # 根据索引获取工作表

table = wb.sheet_by_index(0) # 根据索引获取工作表

table = wb.sheet_by_name(r"三国演义") # 根据名程获取工作表

names = wb.sheet_names() # 获取所有工作表的名字

num = wb.nsheets # 获取所有工作表数量

行[row]操作

total_rows = table.nrows # 获取总行数

current_row_strudata = table.row(0) # 获取该行所有的单元格数据类型和数据组成的列表

current_row_strudata = table.row_slice(0, startcolx=1, endcolx=4)

# 获取该行的列控制[切片]的单元格数据类型和数据组成的列表

current_row_types = table.row_types(3, startcolx=0, endcolx=None)

# 获取该行的列控制[切片]单元格的数据类型组成的列表

current_row_values = table.row_values(4, startcolx=3, endcolx=4) # 获取该行的列控制[切片]单元格的数据组成的列表

current_row_len = table.row_len(0) # 获取该行单元格个数

列[colnum]操作

total_cols = table.ncols # 获取总列数

current_col_strudata = table.col(0, start_rowx=0, end_rowx=4)

# 获取该列的行控制[切片]单元格数据类型和数据组成的列表

current_col_strudata = table.col_slice(0, startrowx=0, endrowx=4) # 获取该列的行控制[切片]单元格数据类型和数据组成的列表

current_col_types = table.col_types(3, startrowx=0, endrowx=None) # 该列的行控制[切片]单元格的数据类型组成的列表

current_col_values = table.col_values(2, startrowx=0, endrowx=1)

# 该列的行控制[切片]单元格的数据组成的列表

cell = table.cell(0, 0) # 获取 单元格对象

cell_type = table.cell_type(2, 3) # 获取单元格数据类型

cell_value = table.cell_value(3, 3) # 获取指定单元格数据

那么这样就可以读取到数据了。