[오늘의 문제] n 번째 원소까지

출처:

def solution(num_list, n):
    answer = []
    
    for i in range(len(num_list)):
        if i == n:
            break       
        answer.append(num_list[i])
        
    return answer

트러블 슈팅

내 기존 코드문엔 엄청 문제가 있었다.

def solution(num_list, n):
    answer = []

    for i in num_list:  
        if i == n:
            break
        answer.append(num_list[i]) 
        
    return answer     

i가 num_list의 원소중 하나인데
그 i 값이 또다시 num_list의 인덱스로 들어가는 셈이다.

자바 알고리즘 하듯이 하는 방법을 채택했다 나는.

인덱스 i 일때 반복문 벗어난다.
그전 까지 num_list[i] 원소를 answer 리스트에 추가시킨다.

다른 사람의 풀이

def solution(num_list, n):
    return num_list[:n]

def solution(num_list, n):
    return [v for i,v in enumerate(num_list) if i<n]