[프로그래머스] level1 : 추억 점수

 

파이썬이랑 친해지기

dictionary 형태의 자료형에 두 리스트의 값을 담기 위해 다음과 같이 작성했다.

name = ["may", "kein", "kain", "radi"]
yearning = [5, 10, 1, 3]
dic = {}
for i, j in zip(name, yearning):
    dic[i] = j

파이썬에서는 iterable한 tuple 값에 대해 dictionary로 초기화해주는 dict() 함수를 제공한다.

 

name = ["may", "kein", "kain", "radi"]
yearning = [5, 10, 1, 3]
dict(zip(name, yearning))

# 실행 결과
# {'may': 5, 'kein': 10, 'kain': 1, 'radi': 3}

코드

def solution(name, yearning, photo):
    dic = dict(zip(name, yearning))
    answer = []
    
    for k in photo:
        score = 0
        for l in k:
            if dic.get(l) is not None:
                score += dic.get(l)
            
        answer.append(score)
    
    return answer