Chunk List

gistfile1.txt

def batch(seq, size):
  return [
    seq[i:i + size]
    for i in range(0, len(seq), size)
  ]


seq = [1,2,3,4,5,6]
size = 3
# OUTPUT: [[1,2,3],[4,5,6]]
print(batch(seq, size))

seq = [1,2,3,4,5]
size = 2
# OUTPUT: [[1,2],[3,4],[5]]
print(batch(seq, size))

seq = [1,2,3]
size = 4
# OUTPUT: [[1,2,3]]
print(batch(seq, size))