프로그래머스 성격 유형 검사하기, 파이썬

문제

https://school.programmers.co.kr/learn/courses/30/lessons/118666?language=python3
MBTI를 구하는 느낌의 문제이다.

코드

def solution(survey, choices):
    types = {"R": 0, "T": 0, "C": 0, "F": 0, "J": 0, "M": 0, "A": 0, "N": 0}
    answer = ''
    for i in range(len(survey)):
        score = choices[i] - 4
        if score > 0: # 동의
            types[str(survey[i][1])] += score
        elif score <= 0: #비동의
            types[str(survey[i][0])] += score * -1
    
    if types["R"] >= types["T"]:
        answer += 'R'
    else:
        answer += 'T'
        
    if types["C"] >= types["F"]:
        answer += 'C'
    else:
        answer += 'F'
        
    if types["J"] >= types["M"]:
        answer += 'J'
    else:
        answer += 'M'
        
    if types["A"] >= types["N"]:
        answer += 'A'
    else:
        answer += 'N'
            
    return answer