Python 字典：运行自定义函数

Question

请协助我完成以下工作。我在 Stack 上遇到了一些类似的问题，但提供的示例并没有解决我遇到的问题。

在下面的字典中，我想在用户键入“选项 1”时运行自定义函数。

在当前设置下，自定义函数 - writeFunction() - 是 无论用户选择哪个选项都会执行。

选择“选项1”时的输出：

• 此功能有效
• 无

选择“选项2”时的输出：

• 此功能有效
• 回答2

如果我将选项 1 更改为字符串值，它会完美执行。我到底做错了什么？

# Custom Function
def writeFunction():
    print("This function works")

# Case statement
def case(arg):
    switch = {
        'Option 1':writeFunction(),
        'Option 2':'Answer 2',
        'Option 3':'Answer 3'
    }
    sysResponse = switch.get(arg,"Value not in list")
    print(sysResponse)

# User selection
userSelection = input("Please select option 1 to 3: ")

# Run case statement based on user selection
case(userSelection)

我想避免使用无休止的 elif 语句。

Answer 1

您可以在dict中引用该函数（注意writeFunction之后缺少()），然后检查您检索到sysResponse的对象是否可调用；如果是，调用它来替换实际值。

这有额外的好处，如果writeFunction 有副作用或执行缓慢，它只会在被选中时执行。

switch = {
    'Option 1': writeFunction,
    'Option 2': 'Answer 2',
    'Option 3': 'Answer 3'
}
sysResponse = switch.get(arg, "Value not in list")
if callable(sysResponse):
    sysResponse = sysResponse()