[Python] list 원하는 갯수대로 자르기

프로그래머스에서 문제를 풀던 중 리스트를 원하는 갯수대로 잘라야하는 상황이 생겼다.

 

예를들어 a = [1,2,3,4,5,6,7,8,9,0] 라고 가정했을 때

원하는 갯수가 2라면

[[1,2], [3,4], [5,6], [7,8], [9,0]] 와 같이 분할하고 싶었다.

 

특정 리스트에서 원하는 갯수대로 요소를 나누는 함수를 만들어보았다.

test = [1,2,3,4,5,6,7,8,9,0]

def list_cut(li, n):
    return [li[i:i+n] for i in range(0, len(li), n)]

print(list_cut(test, 2))
print(list_cut(test, 4))

# 결과값
# [[1, 2], [3, 4], [5, 6], [7, 8], [9, 0]]
# [[1, 2, 3, 4], [5, 6, 7, 8], [9, 0]]