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

문제 설명

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

 

https://school.programmers.co.kr/learn/courses/30/lessons/181865

프로그래머스

제한 사항

0 ≤ a, b ≤ 40,000
0을 제외하고 a, b는 0으로 시작하지 않습니다.

예제

binomial	result
43 + 12	55
0 - 7777	-7777
40000 * 40000	1.6E+09

문제 풀이

def solution(binomial):
    L = binomial.split()
    if L[1] == '+':
        return int(L[0]) + int(L[2])
    elif L[1] == '-':
        return int(L[0]) - int(L[2])
    else:
        return int(L[0]) * int(L[2])

---

파이썬을 독학하시는 분들에게 도움이 되길 바라며,
혹 더 좋은 방법이 있거나 오류가 있다면 편하게 말씀 부탁드립니다.