いろいろ備忘録日記

主に .NET とか Go とか Flutter とか Python絡みのメモを公開しています。

シーケンスを特定数のチャンクに分割

python

どの言語でもちょくちょく使う特定の数のチャンクに分割する処理のPython版。

いつも忘れてしまうのでメモメモ。

def chunks(sequence: Sequence, chunk_size: int = 1) -> Iterator[Any]:
    """
    指定されたシーケンスを指定されたチャンクに分割します.

    :param sequence: シーケンス
    :param chunk_size: チャンクサイズ
    :return: Iterator[Any]
    """
    for i in range(0, len(sequence), chunk_size):
        yield sequence[i:i + chunk_size]

a_list = list(range(100))
for chunk in chunks(a_list, 10):
    print(chunk)
    
[0, 1, 2, 3, 4, 5, 6, 7, 8, 9]
[10, 11, 12, 13, 14, 15, 16, 17, 18, 19]
[20, 21, 22, 23, 24, 25, 26, 27, 28, 29]
[30, 31, 32, 33, 34, 35, 36, 37, 38, 39]
[40, 41, 42, 43, 44, 45, 46, 47, 48, 49]
[50, 51, 52, 53, 54, 55, 56, 57, 58, 59]
[60, 61, 62, 63, 64, 65, 66, 67, 68, 69]
[70, 71, 72, 73, 74, 75, 76, 77, 78, 79]
[80, 81, 82, 83, 84, 85, 86, 87, 88, 89]
[90, 91, 92, 93, 94, 95, 96, 97, 98, 99]

【2017/06/02追記】

numpy.array_split() を利用すればもっと効率よく出来るのを知りました。

import numpy as np

xs = list(range(100))
print(np.array_split(xs, 2))
print(np.array_split(xs, 10))

wordaligned.org

過去の記事については、以下のページからご参照下さい。

サンプルコードは、以下の場所で公開しています。