[Python] 리스트 컴프리헨션을 이용한 리스트 분할

다음과 같은 리스트가 있다고 가정해보자.

lst = [1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12]

갯수가 정확히 나누어 떨어질때

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

# print(solution(lst, 4)) 출력
[[1, 2, 3, 4], [5, 6, 7, 8], [9, 10, 11, 12]]

https://jsikim1.tistory.com/141

Python list chunk - 리스트 분할(자르기, 나누기, split) 방법

 

그렇지 않을때

def solution(data, n):
	return [data[i*n: n*(i+1)] for i in range(0, (len(data)+n-1)//n)]

# print(solution(lst, 5)) 출력
[[1, 2, 3, 4, 5], [6, 7, 8, 9, 10], [11, 12]]

https://kobong.tistory.com/13

Python List를 원하는 개수로 나누기