[프로그래머스] OX퀴즈

• OX퀴즈

문제 설명

덧셈, 뺄셈 수식들이 'X [연산자] Y = Z' 형태로 들어있는 문자열 배열 quiz가 매개변수로 주어집니다. 수식이 옳다면 "O"를 틀리다면 "X"를 순서대로 담은 배열을 return하도록 solution 함수를 완성해주세요.

def solution(quiz):
    quiz_list = []
    answer = []
    for i in quiz:
        quiz_list.append(i.split())
    for i in quiz_list:
        if calculator(i) == True:
            answer.append("O")
        else:
            answer.append("X")
    return answer

def calculator(list):
    result = 0
    answer = int(list[4])
    if list[1] == "+":
        result = int(list[0]) + int(list[2])
    else:
        result = int(list[0]) - int(list[2])
    return result == answer