python-small-examples/md/60.md at master · data-python/python-small-examples

File metadata and controls

19 lines (14 loc) · 283 Bytes

61 不用else和if实现计算器

from operator import *


def calculator(a, b, k):
    return {
        '+': add,
        '-': sub,
        '*': mul,
        '/': truediv,
        '**': pow
    }[k](a, b)


calculator(1, 2, '+')  # 3
calculator(3, 4, '**')  # 81