Python - 根據輸入返回字典鍵

Question

作為預防措施，我對 Python 很陌生。 我正在做一個項目，我在字典中為不同的公司存儲一些不同的變量。 我試圖弄清楚如何從特定字典中返回特定鍵。 例如，用戶輸入他們的公司名稱，程序從他們的特定公司字典中返回一個特定的鍵。

我能夠成功地執行此操作並打印出正確的字符串，但我現在想將此值傳遞給實際調用此代碼但沒有返回任何內容的程序。 我附上了我正在使用的代碼的簡單示例。

Company= "company1"

def getID():

    Companies = {
    "company1" :  {'ID_1': 'id3_12'},
    "company2" :  {'ID_1': 'id="id4_12"', 'ID_2': 'id5_12'},
    "company3" :  {'ID_1': 'id6_12', 'ID_2': 'id7_12'},
    "company4" :  {'ID_1': 'id8_12', 'ID_2': 'id9_12'}
    }


    id = (Companies[str(Company)]['ID_1'])
    return id

例如，有人將輸入“Comany1”，python 將返回“id3_12”

我希望這是有道理的！

謝謝大家先進

Answer 1

您的 function 不是通過參數獲取數據，而是從全局變量Company獲取數據。 如果您不在全局 scope 中（您通常應該避免），那將無法正常工作，試試這個：

def getID(company):
    Companies = {
    "company1" :  {'ID_1': 'id3_12'},
    "company2" :  {'ID_1': 'id="id4_12"', 'ID_2': 'id5_12'},
    "company3" :  {'ID_1': 'id6_12', 'ID_2': 'id7_12'},
    "company4" :  {'ID_1': 'id8_12', 'ID_2': 'id9_12'}
    }
    id = Companies[str(company)]['ID_1']
    return id

print(getID("company1"))