[프로그래머스]OX퀴즈

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

def solution(quiz):
    answer = []
    
    for i in range(len(quiz)):
        re = quiz[i].split(" ")
        if re[1] == "+":
            if int(re[0]) + int(re[2]) == int(re[4]):
                answer.append("O")
            else :
                answer.append("X")
        else : 
            if int(re[0]) - int(re[2]) == int(re[4]):
                answer.append("O")
            else :
                answer.append("X")
    
    return answer