1. ホーム
  2. python

[解決済み】リストをほぼ同じ長さのN個のパーツに分割する

2022-04-15 04:03:14

質問

リストを以下のように分割する最も良い方法は何ですか? おおよそ 等しい部分ですか?例えば、リストが7つの要素を持っていて、それを2つの部分に分割する場合、片方の部分には3つの要素を、もう片方には4つの要素を持たせたいと思います。

のようなものを探しています。 even_split(L, n) を壊すような Ln の部分になります。

def chunks(L, n):
    """ Yield successive n-sized chunks from L.
    """
    for i in range(0, len(L), n):
        yield L[i:i+n]

上のコードは3つのチャンクではなく、3のチャンクを与えています。 単純に転置(これを繰り返し、各列の最初の要素を取り、それをパート1と呼び、次に2番目の要素を取り、それをパート2に入れるなど)することもできますが、それでは項目の順序が崩れてしまいます。

解決方法は?

このコードは丸め誤差のために壊れています。使用しないでください!!!

assert len(chunkIt([1,2,3], 10)) == 10  # fails


ここで、使えそうなものを一つ。

def chunkIt(seq, num):
    avg = len(seq) / float(num)
    out = []
    last = 0.0

    while last < len(seq):
        out.append(seq[int(last):int(last + avg)])
        last += avg

    return out

テスト中です。

>>> chunkIt(range(10), 3)
[[0, 1, 2], [3, 4, 5], [6, 7, 8, 9]]
>>> chunkIt(range(11), 3)
[[0, 1, 2], [3, 4, 5, 6], [7, 8, 9, 10]]
>>> chunkIt(range(12), 3)
[[0, 1, 2, 3], [4, 5, 6, 7], [8, 9, 10, 11]]