Pythonでリストを均等なサイズのチャンクに分割する方法

スライス演算子を使った基本的な方法

リストを均等なサイズのチャンクに分割する最も簡単な方法は、スライス演算子を繰り返し使用し、開始位置と終了位置を固定数ずつずらしていくことです。

次の例では、12個の要素を持つリストを、長さ4の3つのリストに分割しています。

l = [10, 20, 30, 40, 50, 60, 70, 80, 90, 100, 110, 120]
x = 0
y = 12

for i in range(x, y, 4):
    x = i
    print(l[x:x+4])

実行結果：

[10, 20, 30, 40]
[50, 60, 70, 80]
[90, 100, 110, 120]

仕組みの解説

range(x, y, 4)は「0」「4」「8」という値を順番に生成します。各ループ処理でその値を開始位置として l[x:x+4] により4要素分を切り出すことで、元のリストが均等なサイズのチャンクへと分割されます。

リスト内包表記を使った方法

より簡潔に記述したい場合は、リスト内包表記を活用すると1行で実現できます。

l = [10, 20, 30, 40, 50, 60, 70, 80, 90, 100, 110, 120]
chunk_size = 4

chunks = [l[i:i+chunk_size] for i in range(0, len(l), chunk_size)]
print(chunks)
# [[10, 20, 30, 40], [50, 60, 70, 80], [90, 100, 110, 120]]

ジェネレーター関数を使った方法

大量のデータを扱う場合やメモリ効率を重視したい場合は、yield を使ったジェネレーター関数が便利です。全チャンクを一括生成せず、必要なタイミングで順次取り出せます。

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

l = [10, 20, 30, 40, 50, 60, 70, 80, 90, 100, 110, 120]

for chunk in chunk_list(l, 4):
    print(chunk)

まとめ

シンプルな処理ならスライス演算子のループ、可読性を高めたいならリスト内包表記、メモリ効率を重視する大規模データにはジェネレーター関数が適しています。用途に応じて使い分けることで、Pythonでのリスト分割を効率的に行えます。