[프로그래머스] OX퀴즈 - Python

1. 문제

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

프로그래머스

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

 

2. Python code

def solution(quiz):
    answer = []
    for s in quiz:
        a = s.split(' ')
        if eval(''.join(a[:3])) == int(a[-1]):
            answer.append('O')
        else:
            answer.append('X')
    return answer

3. 풀이

python에 있는 eval이란 함수를 사용하면 쉽게 풀 수 있다.

eval 함수는 식을 넣어주면 문자열로 받아서 결과값을 리턴해주는 함수다.

eval("1+2") # 3