Paginieren Sie so oder gibt es einen besseren Algorithmus?

Question

Ich möchte in der Lage sein, eine Sequenz wie folgt zu nehmen:

my_sequence = ['foo', 'bar', 'baz', 'spam', 'eggs', 'cheese', 'yogurt'] 

Verwenden Sie eine Funktion wie:

my_paginated_sequence = get_rows(my_sequence, 3) 

So erhalten Sie:

[['foo', 'bar', 'baz'], ['spam', 'eggs', 'cheese'], ['yogurt']] 

Dies ist, was ich gerade erfunden habe Durchdenken:

def get_rows(sequence, num): 
    count = 1 
    rows = list() 
    cols = list() 
    for item in sequence: 
     if count == num: 
      cols.append(item) 
      rows.append(cols) 
      cols = list() 
      count = 1 
     else: 
      cols.append(item) 
      count += 1 
    if count > 0: 
     rows.append(cols) 
    return rows 

Answer 1

Wenn Sie wissen, dass Sie eine schneidbar Sequenz (Liste oder Tupel),

def getrows_byslice(seq, rowlen): 
    for start in xrange(0, len(seq), rowlen): 
     yield seq[start:start+rowlen] 

Das ist natürlich ein Generator, also wenn Sie unbedingt eine Liste als Ergebnis benötigen, verwenden Sie natürlich list(getrows_byslice(seq, 3)) oder dergleichen.

Wenn das, was Sie beginnen mit einem generischen iterable ist, die itertools recipes bietet Hilfe mit dem grouper Rezepte ...:

import itertools 

def grouper(n, iterable, fillvalue=None): 
    "grouper(3, 'ABCDEFG', 'x') --> ABC DEF Gxx" 
    args = [iter(iterable)] * n 
    return itertools.izip_longest(fillvalue=fillvalue, *args) 

(wieder, werden Sie list dazu aufrufe müssen, wenn eine Liste ist, was du willst natürlich).

Da Sie eigentlich wollen, dass das letzte Tupel abgeschnitten und nicht aufgefüllt wird, müssen Sie die abschließenden Füllwerte vom allerletzten Tupel "trimmen".

Answer 2

Wenn Sie nach gerade nach oben Liste Verständnis sucht, wird dies die Arbeit machen:

L = ['foo', 'bar', 'baz', 'spam', 'eggs', 'cheese', 'yogurt'] 
[L[i*3 : (i*3)+3] for i in range((len(L)/3)+1) if L[i*3 : (i*3)+3]] 
# [['foo', 'bar', 'baz'], ['spam', 'eggs', 'cheese'], ['yogurt']] 
L = ['foo', 'bar', 'baz', 'spam', 'eggs', 'cheese'] 
# [['foo', 'bar', 'baz'], ['spam', 'eggs', 'cheese']] 

Answer 3

Diese Version funktioniert mit jedem (möglicherweise faul und nicht schneidbar) iterable und erzeugt einen faulen iterable (mit anderen Worten, es ist ein Generator ist und arbeitet mit allen Arten von Sequenzen, einschließlich anderen Generatoren):

import itertools 

def paginate(iterable, page_size): 
    while True: 
     i1, i2 = itertools.tee(iterable) 
     iterable, page = (itertools.islice(i1, page_size, None), 
       list(itertools.islice(i2, page_size))) 
     if len(page) == 0: 
      break 
     yield page 

Einige Beispiele:

In [61]: list(paginate(my_sequence, 3)) 
Out[61]: [['foo', 'bar', 'baz'], ['spam', 'eggs', 'cheese'], ['yogurt']] 

In [62]: list(paginate(xrange(10), 3)) 
Out[62]: [[0, 1, 2], [3, 4, 5], [6, 7, 8], [9]] 

Answer 4

Die grouper Funktion in den itertools docs ist klug und prägnant; Das einzige Problem ist, dass Sie die Ergebnisse möglicherweise kürzen müssen, wie Alex Martelli darauf hinwies. Ich würde zu einer Lösung nach Michał Marczyks Antwort neigen, obwohl ich nicht verstehe, warum das nicht viel einfacher gemacht werden kann. Dies funktioniert für alle Fälle, die ich mir vorstellen kann:

import itertools 

def paginate(seq, page_size): 
    i = iter(seq) 
    while True: 
     page = tuple(itertools.islice(i, 0, page_size)) 
     if len(page): 
      yield page 
     else: 
      return