【Python模块】使用Python xlrd读取excel中的测试数据

一、安装xlrd模块

1、安装pip
wget https://bootstrap.pypa.io/get-pip.py
python get-pip.py
2、使用pip安装xlrd
pip install xlrd

二、基本操作

1、获取excel基本信息
data = xlrd.open_workbook('/root/data.xlsx') #打开文件,参数为文件路径
table1 = data.sheets_by_index(0) #通过索引获取表格
table2 = data.sheet_by_name(u'Sheet1') #通过表格名称获取表格
nrows = table2.nrows #获取一共有多少行
ncols =  table2.ncols #获取一共有多少列
row0 = table2.row_values(0) #获取第一行(第一行编号是0)
col0 = table2.col_values(0) #获取第一列(第一列标号是0)
2、封装读取

用封装方法读取表格,得到的结果为一个列表,列表中由多个字典组成。

#!/usr/bin/env python
#coding:utf-8

import xlrd

class GetExcel():
    def __init__(self, path, name):
        self.data = xlrd.open_workbook(path)
        self.table = self.data.sheet_by_name(name)
        self.keys = self.table.row_values(0) #获取第一行的标题
        self.rowNum = self.table.nrows #获取总行数
        self.colNum = self.table.ncols #获取总列数

    def GetData(self):
        if self.rowNum <= 1:
            print ("There's no actual data here, with or without title")
        else:
            r, j = [], 1
            for i in range(self.rowNum-1):
                s, values = {}, self.table.row_values(j)
                for k in range(self.colNum):
                    s[self.keys[k]] = values[k]
                    r.append(s)
                    j += 1
                    return r
if __name__ == '__main__':
    path, name= "/root/data.xlsx", "Sheet1"
    data = GetExcel(path, name)
    print data.GetData() #获得表中的测试数据
3、注意事项

excel中的数据必须是纯文本,打开excel,选中数据,右键-设置单元格格式-文本格式 即可。

三、参考资料

python-excel/xlrd

你可能感兴趣的:(Python模块,Python)