如何在 Python 中的 dictionary.get() 方法中将参数传递给函数（字典项）？

Question

我想用字典替换多个if ... else语句。 以下代码成功运行并输出7 （即第一个参数是'add' ，它调用在my_dict字典中声明的add函数并为其提供参数n1和n2 ）：

def my_calc(operation, n1, n2):

    my_dict = {
        'add': lambda n1, n2: n1 + n2,
        'sub': lambda n1, n2: n1 - n2,
        'mul': lambda n1, n2: n1 * n2,
        'div': lambda n1, n2: n1 / n2
    }

    return my_dict[operation](n1, n2)

print(my_calc('add', 3, 4))

但是，如果我尝试传入一个不熟悉的参数，例如my_calc('abc', 3, 4) ，它会引发KeyError因为abc没有定义。 出于这个原因，我使用了dictionary.get()方法。

如何将参数传递给字典中get()方法返回的函数？ 该函数应返回对 n1 和 n2 执行的操作的结果。 我可以很容易地使用try ... except处理错误，但我想使用dictionary.get()方法来完成，如果操作不在字典中，则让函数返回默认值。

def my_calc(operation, n1, n2):

    my_dict = {
        'add': lambda n1, n2: n1 + n2,
        'sub': lambda n1, n2: n1 - n2,
        'mul': lambda n1, n2: n1 * n2,
        'div': lambda n1, n2: n1 / n2
    }
    
    return my_dict.get(operation(n1, n2), "Default Value")

print(my_calc('abc', 3, 4))

输出：

Traceback (most recent call last):
  File "<string>", line 12, in <module>
File "<string>", line 10, in my_calc
TypeError: 'str' object is not callable

有什么方法可以将参数传递给get()方法内的操作？

项目清单

Answer 1

您的第一个问题是您正在调用操作，它是一个字符串。 您想改为调用get()方法的结果。 这意味着参数在get()括号之外（之后）。

然后，为了提供默认行为， get()中的默认值也应该是一个函数。 它只是不需要对它的论点做任何事情。 您可以像其他人一样使用 lambda。

return my_dict.get(operation, lambda n1, n2: "Default Value")(n1, n2)

如果需要，您可以将默认函数存储在字典中。 这样他们就在一起了。 我们将使用None作为键，因为 1) 有意义（没有操作）并且 2) 用户永远无法输入。

my_dict = {
    None:  lambda n2, n2: "Default value",
    'add': lambda n1, n2: n1 + n2,
    'sub': lambda n1, n2: n1 - n2,
    'mul': lambda n1, n2: n1 * n2,
    'div': lambda n1, n2: n1 / n2
}

然后...

return my_dict.get(operation, my_dict[None])(n1, n2)