12

リストを（入力リストを変更せずに）必要なセグメント長にセグメント化するためのPython組み込み関数（またはメカニズム）を探しています。これが私がすでに持っているコードです：

>>> def split_list(list, seg_length):
...     inlist = list[:]
...     outlist = []
...     
...     while inlist:
...         outlist.append(inlist[0:seg_length])
...         inlist[0:seg_length] = []
...     
...     return outlist
... 
>>> alist = range(10)
>>> split_list(alist, 3)
[[0, 1, 2], [3, 4, 5], [6, 7, 8], [9]]
4

3 に答える 3

22

リスト内包表記を使用できます。

>>> seg_length = 3
>>> a = range(10)
>>> [a[x:x+seg_length] for x in range(0,len(a),seg_length)]
[[0, 1, 2], [3, 4, 5], [6, 7, 8], [9]]
于 2009-08-02T12:26:09.043 に答える
5

出力をどのように使用する必要がありますか? それを反復するだけでよい場合は、グループを生成する反復可能なものを作成することをお勧めします。

def split_by(sequence, length):
    iterable = iter(sequence)
    def yield_length():
        for i in xrange(length):
             yield iterable.next()
    while True:
        res = list(yield_length())
        if not res:
            return
        yield res

使用例:

>>> alist = range(10)
>>> list(split_by(alist, 3))
[[0, 1, 2], [3, 4, 5], [6, 7, 8], [9]]

一度に 1 つのサブセットのみを構築するため、結果をループするだけの場合、これはメモリ内のリスト全体を一度に構築しようとするよりもはるかに少ないメモリを使用します。

>>> for subset in split_by(alist, 3):
...     print subset
...
[0, 1, 2]
[3, 4, 5]
[6, 7, 8]
[9]
于 2009-08-02T12:51:07.353 に答える
2

同じ出力ではありませんが、グルーパー機能が役立つと思います。

from itertools import izip_longest
def grouper(iterable, n, fillvalue=None):
    args = [iter(iterable)] * n
    return izip_longest(*args, fillvalue=fillvalue)

izip_longest を持たない Python2.4 および 2.5 の場合:

from itertools import izip, chain, repeat
def grouper(iterable, n, padvalue=None):
    return izip(*[chain(iterable, repeat(padvalue, n-1))]*n)

いくつかのデモ コードと出力:

alist = range(10)
print list(grouper(alist, 3))

出力: [(0, 1, 2), (3, 4, 5), (6, 7, 8), (9, なし, なし)]

于 2009-08-02T12:38:36.583 に答える