如何从 Python 中的生成器或列表中提取前 N 项？

如何从生成器或列表中检索前 N 个项目？

在 Python 中，有多种方法可以提取前 N 个元素来自列表或生成器。这个答案对不同的方法进行了全面的解释，并与 LINQ 的 Take 方法进行了比较，以供参考。

1.切片列表

与 LINQ 的 Take 方法类似，您可以使用以下语法轻松地对列表进行切片：

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

此切片表示法允许您指定开始和停止索引，以及可选的步长值。您可以省略任何参数来获取部分切片，例如：

• array[start:]
• array[:stop]
• array[::step]

2。对生成器进行切片

与列表不同，生成器在 Python 中不能直接进行切片。相反，您可以使用 itertools.islice() 函数将生成器包装在切片生成器中。语法如下：

<code class="python">import itertools
top5 = itertools.islice(my_list, 5) # grab the first five elements</code>

请记住，对生成器进行切片会部分耗尽它。如果您希望保留整个生成器，请考虑首先将其转换为元组或列表：

<code class="python">result = tuple(generator)</code>

以上是如何从 Python 中的生成器或列表中提取前 N 项？的详细内容。更多信息请关注PHP中文网其他相关文章！