擅长:python、mysql、java
<p>将<a href="https://docs.python.org/3/library/itertools.html#itertools.islice" rel="nofollow noreferrer">itertools.islice</a>与生成器表达式一起使用</p>
<pre><code>from itertools import islice
def get_next_k(curr, k):
return list(islice(curr, k))
arr = [1, 2, 3, 4, 5, 6, 7, 8, 9, 10]
curr = (x for x in arr)
for i in range(4):
print(get_next_k(curr, 3))
</code></pre>
<p><strong>输出:</strong></p>
<pre><code>[1, 2, 3]
[4, 5, 6]
[7, 8, 9]
[10]
</code></pre>