[Python] batch 단위로 iterable 진행

들어가기 앞서,

• 해당 포스트에서는, batch 단위로 iterable 을 진행하는 generator 코드를 작성해보겠습니다.

예제 코드

def batch(iterable, batch_size=1):
  l = len(iterable)
  for ndx in range(0, l, batch_size):                # iterable 데이터를 배치 단위로 확인하며,
    yield iterable[ndx:min(ndx + batch_size, l)]     # batch 단위 만큼의 데이터를 반환

iterable_data = [0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10]    # 데이터 리스트 
batch_size = 3                                        # 배치 사이즈

# iterable 데이터를 배치 단위로 확인하며, batch 단위 만큼의 데이터를 반환
for x in batch(iterable_data, batch_size):
  print(x)

# Output
# [0, 1, 2]
# [3, 4, 5]
# [6, 7, 8]
# [9, 10]

참고

• https://stackoverflow.com/questions/8290397/how-to-split-an-iterable-in-constant-size-chunks