根據輸入從CSV文件中檢索特定行

Question

目前，我有一個程序可以從CSV文件中檢索所有信息，並像這樣輸出：

['USER', 'PASS', 'INFO']
['example', 'pass', 'Apple']
['example1', 'pass1', 'Banana']
['example2', 'pass2', 'Strawberry']

我使用的代碼是：

import csv

username = input("Username... ")

with open('data.csv', 'rt') as f:
reader = csv.reader(f)
for row in reader:
    print(row)

我想做的是對照CSV文件中列出的用戶名檢查用戶輸入，如果匹配，則打印相應的INFO值。 因此，如果用戶名變量為example1，則將打印香蕉。 有沒有辦法做到這一點，類似於MySQL等？

非常感激！

Answer 1

如果您只需要匹配一次，則循環遍歷CSV文件並測試每一行：

with open('data.csv', 'rt') as f:
    reader = csv.reader(f)
    for row in reader:
        if row[0] == username:
            print(row)

如果您需要在代碼中更頻繁地引用不同的用戶名，並且CSV不太大，請使用字典：

with open('data.csv', 'rt') as f:
    reader = csv.reader(f)
    usernames = {r[0]: r for r in reader}

print usernames[username]