使用用户的输入在 excel 文件中搜索特定行（python）

Question

我目前正在制作一个程序来检查哪些用户登录到计算机上，然后根据登录的用户执行不同的操作。因为这是一直在使用“getpass”并且已经成功完成并且用户名是保存到名为“who”的变量中。这是我发现问题的地方，我希望能够在 excel 文件中搜索变量中的名称，并且只打印它所在的行。我该怎么做？

谢谢

Answer 1

您能否将 Excel 文件保存为 CSV 格式？如果是这样，您可以使用 Python 的 csv 模块来解析文件。

假设你可以得到保存为 CSV 格式的文件，名为 directory.csv，它看起来像这样：

first,last,login,phone
Chi,Kaso,ckaso,999-999-9999
Bob,McCormick,bobmac,888-888-8888
Mason,Builder,mbuilder,777-777-7777

下面是打开文件的代码，用 csv.reader 读取每一行，然后找到匹配的行：

import csv
user = 'mbuilder'
with open('directory.csv', mode='r', newline='') as dir_file:
    reader = csv.reader(dir_file)
    for row in reader:
            if row[2] == user:
                    print("Match:", row)
                    break

缺点是它逐行搜索文件，所以如果文件有数百万用户，它会很慢。