Comment récupérer les N premiers éléments des générateurs et des listes en Python ?

Des façons élégantes d'obtenir les N premiers éléments à partir de générateurs ou de listes

Dans LINQ de .NET, récupérer les N premiers éléments d'une collection est simple avec la méthode succincte Take(). Vous vous demandez comment réaliser cet exploit en Python ?

Découper des listes en toute simplicité

Lorsqu'il s'agit de listes, la syntaxe de découpage règne en maître :

top5 = array[:5]

Cette expression simple saisit efficacement les cinq premiers éléments du tableau. Vous pouvez modifier les paramètres pour extraire des plages spécifiques si vous le souhaitez.

Naviguer dans les générateurs avec islice

Les générateurs, cependant, présentent un défi différent. Les découper directement dans Python est une entreprise infructueuse. Entrez itertools.islice(), votre allié de confiance pour les générateurs de découpage :

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

Avertissements astucieux pour les générateurs

N'oubliez pas que l'utilisation d'islice() sur un générateur épuise partiellement il. Pour conserver l'intégralité du générateur intact, envisagez d'abord de le convertir en tuple ou en liste :

result = tuple(generator)

Ce qui précède est le contenu détaillé de. pour plus d'informations, suivez d'autres articles connexes sur le site Web de PHP en chinois!