덧셈, 뺄셈 수식들이 'X [연산자] Y = Z' 형태로 들어있는 문자열 배열 quiz가 매개변수로 주어집니다. 수식이 옳다면 "O"를 틀리다면 "X"를 순서대로 담은 배열을 return하도록 solution 함수를 완성해주세요.
quiz | result |
---|---|
["3 - 4 = -3", "5 + 6 = 11"] | ["X", "O"] |
["19 - 6 = 13", "5 + 66 = 71", "5 - 15 = 63", "3 - 1 = 2"] | ["O", "O", "X", "O"] |
입출력 예 설명
19 - 6 = 13 은 옳은 수식이므로 "O", 5 + 66 = 71 은 옳은 수식이므로 "O", 5 - 15 = 63 은 틀린 수식이므로 "X", 3 - 1 = 2는 옳은 수식이므로 "O" 따라서 ["O", "O", "X", "O"]를 return합니다.
eval
함수 사용하면 문자열로된 식의 계산을 할 수 있으니까 활용해야겠네.1. for q in quiz:
2. e, a = q.split('=')
3. if eval(e) == a:
4. 'O'
5. else:
6. 'X
def valid(q):
e, a = q.split('=') # `=` 기준으로 식과 답 분리
return eval(e) == int(a) # 식이 유효한지 여부
def solution(quiz):
return ['O' if valid(q) else 'X' for q in quiz]
eval
의 활용eval
의 기능을 활용할 수 있다. 수식에서 '='을 '=='으로 수정하면 식의 유효성을 평가해준다.def valid(equation):
equation = equation.replace('=', '==')
return eval(equation)
def solution(equations):
return ["O" if valid(equation) else "X" for equation in equations]