[프로그래머스] 간단한 식 계산하기

 

 

<내가 쓴 코드>

def solution(binomial):
    answer = 0
    a,op,b = binomial.split()   #각 변수를 split으로 나눈다.
    a = int(a)   #a는 문자열이므로 정수로 변환
    b = int(b)   #b도 문자열이므로 정수로 변환
    if op == "+":
        answer = a + b
    elif op == "-":
        answer = a - b
    elif op == "*":
        answer = a * b
    return answer