import itertools

top5 = itertools.islice(array, 5)

W moim guście jest również bardzo zwięzłe łączenie 'zip()' z 'xrange (N) '(lub' range (N) ' w Python3), które działa dobrze na generatorach i wydaje się być bardziej elastyczne dla zmian w ogóle.

# Option #1: taking the first n elements as a list
[x for _, x in zip(xrange(n), generator)]

# Option #2, using 'next()' and taking care for 'StopIteration'
[next(generator) for _ in xrange(n)]

# Option #3: taking the first n elements as a new generator
(x for _, x in zip(xrange(n), generator))

# Option #4: yielding them by simply preparing a function
# (but take care for 'StopIteration')
def top_n(n, generator):
    for _ in xrange(n): yield next(generator)

@Shaikovsky odpowiedź jest doskonała, ale chciałem wyjaśnić kilka punktów.

[next(generator) for _ in range(n)]

Jest to najprostsze podejście, ale rzuca StopIteration, jeśli generator jest przedwcześnie wyczerpany.

Z drugiej strony, następujące podejścia zwracają do n przedmioty, które są prawdopodobnie preferowane w większości sytuacji:

Lista: [x for _, x in zip(range(n), records)]

Generator: (x for _, x in zip(range(n), records))

Odpowiedź jak to zrobić można znaleźć tutaj

>>> generator = (i for i in xrange(10))
>>> list(next(generator) for _ in range(4))
[0, 1, 2, 3]
>>> list(next(generator) for _ in range(4))
[4, 5, 6, 7]
>>> list(next(generator) for _ in range(4))
[8, 9]

Zauważ, że ostatnie połączenie prosi o następne 4, gdy pozostały tylko 2. Użycie list() zamiast {[2] } jest tym, co sprawia, że zrozumienie kończy się na StopIteration wyjątku, który jest wyrzucany przez next().

Masz na myśli pierwsze N elementy, czy N największe elementy?

Jeśli chcesz pierwszy:

top5 = sequence[:5]

Działa to również dla największych N pozycji, zakładając, że Twoja sekwencja jest sortowana w porządku malejącym. (Twój przykład LINQ wydaje się również zakładać to.)

Jeśli chcesz największy, a nie jest posortowany, najbardziej oczywistym rozwiązaniem jest najpierw posortowanie:

l = list(sequence)
l.sort(reverse=True)
top5 = l[:5]

Aby uzyskać bardziej wydajne rozwiązanie, użyj sterty min (dzięki Thijs):

import heapq
top5 = heapq.nlargest(5, sequence)

Z itertools uzyskasz kolejny obiekt generatora, więc w większości przypadków będziesz potrzebował kolejnego kroku, aby wykonać pierwsze N elementów (N). Istnieją co najmniej dwa prostsze rozwiązania (nieco mniej wydajne pod względem wydajności, ale bardzo poręczne), aby przygotować elementy do użycia z generator:

Użycie rozumienia listy:

first_N_element=[generator.next() for i in range(N)]

Inaczej:

first_N_element=list(generator)[:N]

Gdzie N jest liczbą elementów, które chcesz wziąć (np. N = 5 dla pierwszych pięciu elementów).

To powinno zadziałać

top5 = array[:5]