基于变量的决策传递给函数

Question

我是一名新的Python 3用户，需要以某种方式进行以下工作。我不知道如何根据变量operation正确地采取行动。

def arith(x, y, operation):
    if operation == add:
        return x + y
    if operation == mult:
        return x * y
print(arith(x = 2, y = 3, operation = add))
print(arith(x = 2, y = 3, operation = mult))

我收到以下错误消息：

print（arith（x = 2，y = 3，operation = add）） NameError：未定义名称“add”

我看过但找不到这类问题的答案

Answer 1

def arith(x, y, operation):
    if operation == 'add':
        return x + y
    if operation == 'mult':
        return x * y
arith(2, 3, 'add')
arith(2, 3, 'mult')

问题是没有定义add和mult。通过将它们用引号括起来，'add'和'mult'，你可以将它们定义为字符串。

查看有关字符串的this文档以获取更多信息。