[Level 1] 모의고사(enumerate)

def solution(answers):
    supo_1 = [1, 2, 3, 4, 5]
    supo_2 = [2, 1, 2, 3, 2, 4, 2, 5]
    supo_3 = [3, 3, 1, 1, 2, 2, 4, 4, 5, 5]
    supo_count = [0, 0, 0]

    for i in range(len(answers)):
        if answers[i] == supo_1[i % 5]:
            supo_count[0] += 1
        if answers[i] == supo_2[i % 8]:
            supo_count[1] += 1
        if answers[i] == supo_3[i % 10]:
            supo_count[2] += 1

    k = max(supo_count)
    answer = []

    if supo_count[0] == k:
        answer.append(1)
    if supo_count[1] == k:
        answer.append(2)
    if supo_count[2] == k:
        answer.append(3)

    return answer

enumerate를 활용한 코드

• enumertate: 인덱스와 원소를 동시에 접근하면서 루프를 돌리는 함수. Pythonic한 함수!

def solution(answers):
    pattern1 = [1,2,3,4,5]
    pattern2 = [2,1,2,3,2,4,2,5]
    pattern3 = [3,3,1,1,2,2,4,4,5,5]
    score = [0, 0, 0]
    result = []

    for idx, answer in enumerate(answers):
        if answer == pattern1[idx%len(pattern1)]:
            score[0] += 1
        if answer == pattern2[idx%len(pattern2)]:
            score[1] += 1
        if answer == pattern3[idx%len(pattern3)]:
            score[2] += 1

    for idx, s in enumerate(score):
        if s == max(score):
            result.append(idx+1)

    return result

# 일반적인 for문
for 원소 in 목록:
	print(원소)

# 인덱스를 함께 출력하고 싶은 경우
for i in range(len(목록)):
	print(i, 목록[i])
    
# enumerate를 활용하여 인덱스와 값 함께 출력
for i, 원소 in enumerate(목록):
	print(i, 원소)