如何使用用户输入来调用用户定义的函数？

Question

我有以下功能

def retrieve_phone_data():
    # function does something
    return phone_list

def retrieve_customer_data():
    # function does something
    return customer_info

def main(input1, input2):
    if input1 == 'phone' and input2 == 'customer':
        retrieve_phone_data()
        retrieve_customer_data()
    else:
        print('Please enter valid input')

input1 和 input2 只能有特定的值（让我们说 "phone" 和 "customer" 现在）。我有 10 多个函数，我不想使用 If 语句，因为它太多了。

我需要像 excel 中的 vlookup 这样的东西，我将创建可能的输入列表和共同赞助函数，以便每个有效的用户输入都将链接到该列表中的一个函数。有什么办法吗？

Answer 1

您可以在此处使用dict：

data_retrievers = {
    "phone": retrieve_phone_data,
    "customer": retrieve_customer_data,
}

然后在你的 main 方法中：

if input in data_retrievers:
    data_retriever = data_retrievers[input]
    retrieved_data = data_retriever()
else:
    # Signify invalid input

如果要添加另一个输入，只需在data_retrievers 中添加一个条目，键为input，值为对应的函数。注意data_retrievers中函数名后面缺少()，该函数仅在获取data_retriever时调用。