python-3.x - 只需要知道我使用operator模块的方式，下面程序中python的操作符函数是否合适？

Question

下面的程序接受用户的选择，（add，sub，mul，div）然后根据用户输入的2个数字的选择执行操作。只需要知道我使用运算符函数作为 dict 值来完成任务的方式是否合适？我可以使用 lambda 函数作为 dict 值来代替这些运算符函数吗？

import operator
def opnnum(opn, x, y):
    d ={"add" : operator.add,
        "mul":operator.mul,
        "div":operator.truediv,
        "sub":operator.sub}
    return d[opn](x,y)

op = input("Enter add for +, sub for -, mul for * and div for /")
n1 = int(input("Enter first number"))
m1 = int(input("Enter second number"))
print("The answer of operation",op, "is",opnnum(op, n1, m1))

标签： python-3.x