¿Cómo tomar los primeros N elementos de un generador o lista en Python?

Question

Con linq Me

var top5 = array.Take(5); 

Cómo hacer esto con Python?

Answer 1

rebanar una lista

top5 = array[:5] 

• para cortar una lista, hay una sintaxis simple: array[start:stop:step]
• Puede omitir cualquier parámetro. Estas son todas válidas: array[start:], array[:stop], array[::step]

Cortando un generador

import itertools 
top5 = itertools.islice(my_list, 5) # grab the first five elements 

• No se puede cortar un generador directamente en Python. itertools.islice() envolverá un objeto en un nuevo generador de división utilizando la sintaxis itertools.islice(generator, start, stop, step)

• Recuerde, cortar un generador lo agotará parcialmente.Si desea mantener intacto todo el generador, tal vez convertirla en una tupla o lista de principio, como: result = tuple(generator)

Answer 2

import itertools 

top5 = itertools.islice(array, 5) 

Answer 3

Esto debería funcionar

top5 = array[:5] 

Answer 4

¿Quiere decir que los primeros N elementos, o los N mayores artículos?

Si desea que la primera:

top5 = sequence[:5] 

Esto también funciona para los mayores elementos N, en el supuesto de que su secuencia se ordena en orden descendente. (. Su ejemplo LINQ parece asumir esto también)

Si desea que el más grande, y que no está ordenada, la solución más obvia es ordenar primero:

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

Para una más performante solución, utilizar un min-heap (gracias Thijs):

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

Answer 5

Usted tiene que usar rebanadas:

Prueba esto:

>>> lst = [1,2,3,4,5] 

>>> lst[:2] 
[1, 2] 

Toma de índice 0 al índice 2.

También puede hacer cosas como:

>>> lst = [1,2,3,4,5] 

>>> lst[2:4] 

[3, 4] 

Answer 6

En mi gusto, también es muy concisa para combinar 'zip()' con 'xrange (n) '(o' rango (n) 'en Python3), que también funciona bien en los generadores y parece ser más flexible para los cambios en general.

# 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) 

Answer 7

Con itertools obtendrá otro objeto generador, así que en la mayoría de los casos será necesario un paso más al dar los primeros N elementos (N). Hay al menos dos soluciones más simples (un poco menos eficiente en términos de rendimiento, pero muy práctico) para obtener los elementos listos para usar de una generator:

Utilizando lista por comprensión:

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

lo contrario:

first_N_element=list(generator)[:N] 

Donde N es el número de elementos que desea tomar (por ejemplo, N = 5 para los primeros cinco elementos).

Answer 8

La respuesta para saber cómo hacer esto se puede encontrar here

>>> 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] 

en cuenta que la última llamada para la siguiente pregunta 4, cuando sólo 2 están quedando. El uso de list() en lugar de [] es lo que hace que la comprensión finalice en la excepción StopIteration lanzada por next().

Answer 9

@ La respuesta de Shaikovsky es excelente, pero quería aclarar un par de puntos.

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

Este es el enfoque más simple, pero los tiros StopIteration si el generador es prematuramente agotado.

---

Por otro lado, los siguientes enfoques volver hastan artículos, que es sin duda preferible en la mayoría de circunstancias:

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

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