Comment récupérer les N premiers éléments d'un générateur ou d'une liste en Python ?

Comment récupérer les N premiers éléments d'un générateur ou d'une liste ?

Similaire à l'expression LINQ "var top5 = array.Take( 5)," comment pouvons-nous accomplir la même chose en Python ?

Découper une liste

Python fournit une syntaxe concise pour découper des listes :

<code class="python">top5 = array[:5]</code>

• Les paramètres représentent le début, l'arrêt et l'étape.
• Tout paramètre peut être omis, ce qui donne lieu à des expressions valides telles que array[start:], array[:stop] ou array[::step ].

Découper un générateur

Les générateurs ne peuvent pas être découpés directement en Python. Utilisez plutôt itertools.islice(), qui crée un nouveau générateur de découpage :

<code class="python">import itertools
top5 = itertools.islice(my_list, 5)</code>

• N'oubliez pas que le découpage d'un générateur l'épuise partiellement. Si vous devez conserver l'intégralité du générateur, convertissez-le au préalable en tuple ou en liste, par exemple, 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!