본문 바로가기

코딩/프로그래머스

[프로그래머스 / 파이썬] 리스트 자르기

728x90
반응형

문제 설명

정수 n 정수 3개가 담긴 리스트 slicer 그리고 정수 여러 개가 담긴 리스트 num_list 주어집니다slicer 담긴 정수를 차례대로 a, b, c라고 n 따라 다음과 같이 num_list 슬라이싱 하려고 합니다.

  • n = 1 : num_list 0 인덱스부터 b 인덱스까지
  • n = 2 : num_list a 인덱스부터 마지막 인덱스까지
  • n = 3 : num_list a 인덱스부터 b 인덱스까지
  • n = 4 : num_list a 인덱스부터 b 인덱스까지 c 간격으로

올바르게 슬라이싱한 리스트를 return하도록 solution 함수를 완성해주세요.


제한사항

  • n  1, 2, 3, 4 하나입니다.
  • slicer 길이 = 3
  • slicer 담긴 정수를 차례대로 a, b, c라고
  • 5 ≤ num_list 길이 ≤ 30
  • 0 ≤ num_list 원소 ≤ 100

입출력

n slicer num_list result
3 [1, 5, 2] [1, 2, 3, 4, 5, 6, 7, 8, 9] [2, 3, 4, 5, 6]
4 [1, 5, 2] [1, 2, 3, 4, 5, 6, 7, 8, 9] [2, 4, 6]

입출력 설명

입출력 #1

  • [1, 2, 3, 4, 5, 6, 7, 8, 9]에서 1 인덱스부터 5 인덱스까지 자른 리스트는 [2, 3, 4, 5, 6]입니다.

입출력 #2

  • [1, 2, 3, 4, 5, 6, 7, 8, 9]에서 1 인덱스부터 5 인덱스까지 2 간격으로 자른 리스트는 [2, 4, 6]입니다.

 


solution.py

def solution(n, slicer, num_list):
    answer = []
    
    if n == 1:
        answer = num_list[:slicer[1]+1]
    if n == 2:
        answer = num_list[slicer[0]:]
    if n == 3:
        answer = num_list[slicer[0]:slicer[1]+1]
    if n == 4:
        answer = num_list[slicer[0]:slicer[1]+1:slicer[2]]
    return answer



반응형