Python读取Excel数据并根据列名取值

一直想将自己接触到的东西梳理一遍，可就是迈不出第一步，希望从这篇总结开始不要再做行动的矮人了。

最近测试过程中需要用到python读取excel用例数据，于是去了解和学习了下xlrd库，这里只记录使用过程中读取excel数据相关操作。

一、安装xlrd库

　　可以下载xlrd库包到本地安装，也可以通过pip命令安装，这里我选择pip命令：

　　pip install xlrd

二、使用xlrd读取excel数据

　　具体详细的操作可以参考xlrd库操作说明文档，以下是两种读取excel数据的方法：

　　1、根据Excel中sheet名称读取数据：　　　　

 1 def readExcelDataByName(fileName, sheetName):
 2     table = None
 3     errorMsg = None
 4     try:
 5         data = xlrd.open_workbook(fileName)
 6         table = data.sheet_by_name(sheetName)
 7     except Exception, msg:
 8         errorMsg = msg  
 9     return table, errorMsg

　　2、根据Excel中sheet的序号获取：

1 def readExcelDataByIndex(fileName, sheetIndex):
2     table = None
3     errorMsg = ""
4     try:
5         data = xlrd.open_workbook(fileName)
6         table = data.sheet_by_index(sheetIndex)
7     except Exception, msg:
8         errorMsg = msg
9     return table, errorMsg

　　3、根据列名获取相应序号，由于有时读取excel中列数据时，需要通过列头名称获取相应的列中的值，所以写了下面这个返回列名所在表格中的index。然后就可以直接通过table.cell_value(i, getColumnIndex(table,'列名'))获取列的值。

 1 def getColumnIndex(table, columnName):
 2     columnIndex = None 3     
 3     for i in range(table.ncols): 5        
 4         if(table.cell_value(0, i) == columnName):
 5             columnIndex = i
 6             break
 7     return columnIndex

　　下面加入需要读取如下excel表格中的数据,在读取数据时直接根据列名去获取相应的值。

根据列名读取相应的值，代码如下：

 1 #!/usr/bin/python
 2 # coding=utf-8
 3 __author__ = 'Paul'
 4 import xlrd
 5 import chardet
 6 import traceback
 7 def getColumnIndex(table, columnName):
 8     columnIndex = None
 9     #print table
10     for i in range(table.ncols):
11         #print columnName
12         #print table.cell_value(0, i)
13         if(table.cell_value(0, i) == columnName):
14             columnIndex = i
15             break
16     return columnIndex
17 def readExcelDataByName(fileName, sheetName):
18     #print fileName
19     table = None
20     errorMsg = ""
21     try:
22         data = xlrd.open_workbook(fileName)
23         table = data.sheet_by_name(sheetName)
24     except Exception, msg:
25         errorMsg = msg
26     return table, errorMsg
27 def readExcelDataByIndex(fileName, sheetIndex):
28     table = None
29     errorMsg = ""
30     try:
31         data = xlrd.open_workbook(fileName)
32         table = data.sheet_by_index(sheetIndex)
33     except Exception, msg:
34         errorMsg = msg
35     return table, errorMsg
36 if __name__ == '__main__':
37     #example
38     xlsfile= 'F:/test_AutoTesting/TestCase/RunList.xlsx'
39     table = readExcelDataByName(xlsfile, 'Sheet1')[0]
40     #获取第一行的值    
41     testcase_id = table.cell_value(1, getColumnIndex(table,'TestCaseID'))
42     app_config = table.cell_value(1, getColumnIndex(table,'APPConfig'))
43 
44     print u'测试用例ID为：%s'%(testcase_id)
45     print u'配置信息为：%s'%(app_config)

得出结果如下：

　　4、读取excel中的文本或数值转换成了float的问题

　　　有时Excel中的值为20，但读取出来的值却变成了20.0，这与我们想要的不大一致，特别是做UI自动化测试过程中需要下拉选择值时就完全选不出想要的选项了。目前我想到的是通过下面的语句来处理：

if isinstance(inputValue,float):    #判断读取到的值是否为float
    if inputValue==int(inputValue): #判断读取到的值与转成int后的值是否相等，如果相等则转成int
    inputValue = int(inputValue)
    inputValue = str(inputValue) #转成str