如何从生成器或列表中检索前 N 个项目?
在 Python 中,有多种方法可以提取前 N 个元素来自列表或生成器。这个答案对不同的方法进行了全面的解释,并与 LINQ 的 Take 方法进行了比较,以供参考。
1.切片列表
与 LINQ 的 Take 方法类似,您可以使用以下语法轻松地对列表进行切片:
<code class="python">top5 = array[:5]</code>
此切片表示法允许您指定开始和停止索引,以及可选的步长值。您可以省略任何参数来获取部分切片,例如:
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中文网其他相关文章!