간단한 식 계산하기

문자열 binomial이 매개변수로 주어집니다.
binomial은 "a op b" 형태의 이항식이고
a와 b는 음이 아닌 정수, op는 '+', '-', '*' 중 하나입니다.
주어진 식을 계산한 정수를 return 하는 solution 함수를 작성해 주세요.

내 코드

def solution(binomial):
    answer = 0
    if '+' in binomial:
        a,b = map(int,binomial.split('+'))
        answer = a+b
    elif '-' in binomial:
        a,b = map(int,binomial.split('-'))
        answer = a-b
    elif '*' in binomial:
        a,b = map(int,binomial.split('*'))
        answer = a*b
    return answer

예시 코드

def solution(binomial):
    a, op, b = binomial.split()

    a = int(a)
    b = int(b)

    if op == "+":
        result = a + b
    elif op == "-":
        result = a - b
    elif op == "*":
        result = a * b

    return result