根据输入从excel读取特定行

时间:2018-04-06 03:51:14

标签: python

我想根据用户输入从Excel工作表中读取一行。我的Excel工作表包含具有编号,优先级,受让人和其他详细信息的列。 例如: - 如果我想获取number = PRB00000的整行数据,我目前正在使用下面的代码。我正在使用一个列表并手动查找它的索引并将其硬编码为VARIABLE,然后将其发送到selenium webdriver,例如: - webelement.send_keys(变量)。我不想这样做,而是根据用户输入读取行并将值存储在各自的列名称中。

book=xlrd.open_workbook('C:\\xxxx\\xxxx\\third party power.xlsx')
#print (book.nsheets)
#print (book.sheet_names())

first_sheet=book.sheet_by_index(0)
prb= "PRB0045087"
list1 = []
for sheet in book.sheets():
 for rowidx in range(sheet.nrows):
     row=sheet.row(rowidx)
     for colidx, cell in enumerate(row):
         if cell.value == prb :
             list1 = first_sheet.row_values(rowidx)

   RCA = list1[20]
   LTF = list1[21]

1 个答案:

答案 0 :(得分:0)

建议你做一个ID和行索引的字典,这样就可以查找了

ids = {cell.value:colidx for colidx,cell in enumerate(row)}