[Coding Test/Python][Programmers] n 번째 원소까지

문제 설명

정수 리스트 num_list와 정수 n이 주어질 때, num_list의 첫 번째 원소부터 n 번째 원소까지의 모든 원소를 담은 리스트를 return하도록 solution 함수를 완성해주세요.

제한사항

2 ≤ num_list의 길이 ≤ 30
1 ≤ num_list의 원소 ≤ 9
1 ≤ n ≤ num_list의 길이 ___

 

초기 코드

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

직접 하나씩 꺼내서 추가하고 과정이 눈에 보인다는 점에서 초보자가 이해하기 쉬운 코드다.

하지만 슬라이싱을 활용한다면 훨씬 간결하게 풀이가 가능하다. (실무 사용 방식)

 

최종 코드

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

슬라이싱의 경우 반복문에서 발생할 수 있는 인덱스 오류를 사전에 방지할 수 있다.

따라서 더 안전하고 간단하게 표현이 가능하다. (더 파이썬적인 사고랄까..)