如何使用 Python 從文件中讀取數據？

Question

我正在尋找查找 function 以打印相應的名稱。 我有一個列表文件

a Chief manager
b Assistant general manager
c general manager
D CTO
E CEO

我的腳本中有一個變量“用戶”我想檢查第一列中的變量，如果值匹配則打印相應的名稱

例如，如果“user”==a 則打印“Chief manager”

Answer 1

使用字典會更容易。

d = { 'a': 'Chief Manager', 'b': 'Assistant General Manager', 'c': 'General Manager', 'D': 'CTO', 'E': 'CEO' }
user = 'a'
print(d[user])
# Chief Manager

要將該文件加載到字典中，您可以執行以下操作：

with open('/path/to/my/file') as myFile:
    d = dict(line.strip().split(None, 1) for line in myFile)

print(d['a'])
# Cheif manager

---

如果要解析 CSV 文件，則取決於文件的實際格式。 如果它看起來像這樣：

a,"Cheif manager"
b,"Assistant general manager"
c,"general manager"
D,"CTO"
E,"CEO"

你可以這樣讀：

from csv import reader
d = dict(row for row in reader(open('/path/to/my/file'), delimiter=',', quotechar='"'))
print(d['a'])
# Cheif manager

Answer 2

假設第一列在某種程度上是唯一的，您的第一個任務是使用第一列的鍵和第二列的值構建一個字典。

要構建字典，請按以下方式嘗試。

org_dict = dict()
line = 'a Chief manager'
key,value = line.split(None,1)
org_dict[key] = value

現在要從文件中獲取每一行，您可以打開文件並逐行讀取。

with open('myfile') as f:
    for line in f:
        # process your line