[python] 파이썬 리스트에서 원소들을 원하는 개수로 분할하는 방법은?
def chunk(input_list, chunk_size):
return [input_list[i:i + chunk_size] for i in range(0, len(input_list), chunk_size)]
위의 코드는 input_list
라는 리스트를 chunk_size
개수로 분할하여 새로운 리스트를 반환합니다. 리스트를 분할할 때에는 range
를 사용하여 각 청크의 시작 인덱스를 계산하고, 리스트 슬라이싱을 통해 원하는 개수로 분할합니다.
이제, 다음과 같이 함수를 호출하여 리스트를 분할할 수 있습니다.
my_list = [1, 2, 3, 4, 5, 6, 7, 8, 9, 10]
chunked_list = chunk(my_list, 3)
print(chunked_list)
위 코드를 실행하면 [1, 2, 3], [4, 5, 6], [7, 8, 9], [10]
와 같이 원하는 개수로 리스트가 분할된 결과를 얻을 수 있습니다.
참고 문헌:
- 파이썬 공식 문서: https://docs.python.org/3/library/functions.html#func-range