用Python读取Excel

Question

我有一个函数需要两个参数：user和password，它们都是字符串。我的问题是，我想从Excel文件（用户和密码）中获取参数。我该怎么做（

该函数如下所示（我将Selenium与Chrome结合使用）：

def login(user,passwd):
    createDir()
    driver.get('https://www.fasab2b.com/FasaCL/BBRe-commerce/access/login.do')
    driver.maximize_window()
    txtId = driver.find_element_by_id("logid")
    passw = driver.find_element_by_id("password")
    txtId.send_keys(user)
    passw.send_keys(passwd)
    passw.send_keys(Keys.RETURN)

我想用如下方式执行函数：

for user,password in excelFile:
    login(user,password)
    print("login made with: " + user + password)

谢谢你的帮助

Answer 1

我建议openpyxl

这是如何从现有Excel文件访问工作表

import openpyxl

wb = openpyxl.load_workbook('data/test.xlsx')
ws = wb.worksheets[0]

因此，创建一个函数非常容易，例如，它返回一个元组列表，该列表来自给定Excel文件的所有行（从第二行，跳过标题，到最后一行）和特定列（在您的示例中也是3和4）

def excelLogins(path):
    wb = openpyxl.load_workbook(path)
    ws = wb.worksheets[0]
    ret = []
    for i in range(2,ws.max_row + 1):
        ret.append((ws.cell(i,3).value,str(ws.cell(i,4).value)))
    return ret

由于您的密码将被解释为Excel中的数字，您可能还希望将其转换为字符串，如上所述。 最终，您可以根据自己的喜好打印结果

for user,password in excelLogins('data/test.xlsx'):
    print("login made with: " + user + ':' + password)