파이썬 리스트를 동일한 크기의 청크로 나누는 방법

2024-08-13

문제 이해:

파이썬에서 하나의 리스트를 여러 개의 동일한 크기의 작은 리스트들로 나누는 것을 의미합니다. 예를 들어, 10개의 요소를 가진 리스트를 2개의 청크로 나누면 각 청크는 5개의 요소를 갖게 됩니다.

해결 방법:

for 루프를 이용한 직접 구현:

가장 기본적인 방법으로, for 루프를 이용하여 리스트를 순회하며 원하는 크기만큼 요소를 추출하여 새로운 리스트에 추가하는 방식입니다.

def split_list(list_to_split, chunk_size):
  """
  주어진 리스트를 동일한 크기의 청크로 나눕니다.

  Args:
    list_to_split: 나눌 리스트
    chunk_size: 각 청크의 크기

  Returns:
    나눠진 청크들의 리스트
  """

  result = []
  for i in range(0, len(list_to_split), chunk_size):
    result.append(list_to_split[i:i+chunk_size])
  return result

# 예시
my_list = [1, 2, 3, 4, 5, 6, 7, 8, 9, 10]
chunk_size = 3
result = split_list(my_list, chunk_size)
print(result)  # [[1, 2, 3], [4, 5, 6], [7, 8, 9], [10]]

list comprehension을 이용한 간결한 구현:

list comprehension을 사용하면 위의 코드를 더욱 간결하게 작성할 수 있습니다.

def split_list(list_to_split, chunk_size):
  return [list_to_split[i:i+chunk_size] for i in range(0, len(list_to_split), chunk_siz   e)]

itertools.islice를 이용한 효율적인 구현:

itertools 모듈의 islice 함수를 사용하면 더욱 효율적인 방법으로 리스트를 나눌 수 있습니다.

from itertools import islice

def split_list(iterable, size):
    it = iter(iterable)
    while True:
        chunk = list(islice(it, size))
        if not chunk:
            return
        yield chunk

# 예시
my_list = [1, 2, 3, 4, 5, 6, 7, 8, 9, 10]
chunk_size = 3
result = list(split_list(my_list, chunk_size))
print(result)

어떤 방법을 선택해야 할까요?

가독성: for 루프를 이용한 방법이 가장 직관적이지만, list comprehension이나 itertools.islice를 사용하면 더욱 간결하게 표현할 수 있습니다.
효율성: itertools.islice를 사용하는 방법이 일반적으로 가장 효율적입니다.
활용도: 문제 상황에 따라 적절한 방법을 선택해야 합니다. 예를 들어, 생성기(generator)를 사용해야 하는 경우 itertools.islice가 유용합니다.

주의 사항:

리스트의 길이가 청크 크기로 나누어 떨어지지 않을 경우, 마지막 청크에는 나머지 요소들이 포함될 수 있습니다.
매우 큰 리스트를 다룰 때는 메모리 사용량에 주의해야 합니다.

응용:

데이터 처리: 큰 데이터셋을 작은 단위로 나누어 처리할 때 유용합니다.
병렬 처리: 각 청크를 다른 프로세스나 스레드에서 처리하여 병렬 처리를 수행할 수 있습니다.
파일 처리: 큰 파일을 일정한 크기의 청크로 나누어 읽거나 쓸 때 사용할 수 있습니다.

파이썬 리스트를 동일한 크기의 청크로 나누는 다양한 예시 코드

앞서 설명한 세 가지 방법 외에도, 더욱 다양한 상황에 맞춰 활용할 수 있는 샘플 코드들을 살펴보겠습니다.

numpy 라이브러리 활용:

numpy 라이브러리를 사용하면 간단하게 배열을 분할할 수 있습니다.

import numpy as np

my_list = [1, 2, 3, 4, 5, 6, 7, 8, 9, 10]
chunk_size = 3

# 리스트를 numpy 배열로 변환
arr = np.array(my_list)

# 배열을 reshape하여 원하는 크기의 행렬 생성
result = np.reshape(arr, (-1, chunk_size))

# 다시 리스트로 변환
result = result.tolist()
print(result)  # [[1, 2, 3], [4, 5, 6], [7, 8, 9], [10]]

numpy를 사용하면 벡터화 연산을 통해 빠른 처리가 가능하며, 다차원 배열을 다룰 때 유용합니다.

pandas 라이브러리 활용:

pandas 라이브러리를 사용하면 데이터 분석에 특화된 기능들을 활용하여 리스트를 분할할 수 있습니다.

import pandas as pd

my_list = [1, 2, 3, 4, 5, 6, 7, 8, 9, 10]
chunk_size = 3

# 리스트를 Series로 변환
series = pd.Series(my_list)

# chunk_size 크기로 나누어 DataFrame 생성
df = pd.DataFrame(series)

# 각 행을 리스트로 변환
result = df.values.tolist()
print(result)  # [[1], [2], [3], [4], [5], [6], [7], [8], [9], [10]]

pandas는 데이터 분석에 특화된 라이브러리이므로, 데이터 분석 작업과 함께 리스트 분할을 수행할 때 유용합니다.

모듈화된 함수 생성:

자주 사용하는 기능은 함수로 만들어 재사용성을 높일 수 있습니다.

def split_list_into_chunks(data, chunk_size):
    """
    주어진 리스트를 동일한 크기의 청크로 나누는 함수

    Args:
        data: 나눌 리스트
        chunk_size: 각 청크의 크기

    Returns:
        나눠진 청크들의 리스트
    """

    for i in range(0, len(data), chunk_size):
        yield data[i:i + chunk_size]

# 사용 예시
my_list = [1, 2, 3, 4, 5, 6, 7, 8, 9, 10]
chunk_size = 3
result = list(split_list_into_chunks(my_list, chunk_size))
print(result)

numpy: 수치 계산 및 배열 연산에 특화되어 빠른 처리가 필요할 때
pandas: 데이터 분석 작업과 함께 리스트 분할을 수행할 때
일반적인 리스트 처리: for 루프, list comprehension, itertools.islice 등을 활용
재사용성: 함수로 모듈화하여 코드를 재사용

추가 고려 사항:

메모리 효율성: 매우 큰 리스트를 다룰 때는 메모리 사용량을 줄이기 위해 생성기를 사용하는 방법을 고려할 수 있습니다.
데이터 타입: 리스트 외에도 튜플, 문자열 등 다양한 데이터 타입에 적용할 수 있습니다.
불규칙한 크기의 청크: 필요에 따라 불규칙한 크기의 청크로 나눌 수 있는 알고리즘을 구현할 수 있습니다.

파이썬 리스트를 동일한 크기의 청크로 나누는 다양한 방법

앞서 살펴본 방법 외에도 다양한 상황에 맞춰 활용할 수 있는 몇 가지 추가적인 방법들을 소개해 드리겠습니다.

재귀 함수를 이용한 구현:

재귀 함수를 활용하여 리스트를 반으로 나누는 방식으로 청크를 생성할 수 있습니다.

def split_list_recursive(lst, n):
  """
  재귀 함수를 이용하여 리스트를 동일한 크기의 청크로 나눕니다.

  Args:
    lst: 나눌 리스트
    n: 각 청크의 크기

  Returns:
    나눠진 청크들의 리스트
  """

  if not lst:
    return []
  return [lst[:n]] + split_list_recursive(lst[n:], n)

generator expression을 이용한 간결한 구현:

generator expression을 이용하여 더욱 간결하게 코드를 작성할 수 있습니다.

def split_list_generator(lst, n):
  """
  generator expression을 이용하여 리스트를 동일한 크기의 청크로 나눕니다.

  Args:
    lst: 나눌 리스트
    n: 각 청크의 크기

  Yields:
    각 청크
  """

  for i in range(0, len(lst), n):
    yield lst[i:i+n]

외부 라이브러리 more_itertools 활용:

more_itertools 라이브러리의 grouper 함수를 사용하면 간단하게 리스트를 청크로 나눌 수 있습니다.

from more_itertools import grouper

my_list = [1, 2, 3, 4, 5, 6, 7, 8, 9, 10]
chunk_size = 3

result = list(grouper(my_list, chunk_size))
print(result)  # [(1, 2, 3), (4, 5, 6), (7, 8, 9), (10, None, None)]