Python itertools.islice 怎样截取迭代器而不提前展开

我在处理一批按行到达的日志时，最不想做的一件事，就是先把整个生成器变成列表，再取中间那几十行。Python 的 itertools.islice 正好解决这个问题：它返回一个新的迭代器，用 start、stop、step 描述要保留的位置，不要求输入先整体展开。

如果数据源本来就是列表，列表切片更直观；如果数据源是文件、生成器或分页流，islice 通常更合适。关键是先分清“序列索引”和“迭代器消费”这两套语义。

本文先记住三点：窗口是惰性的；参数只能使用非负位置；完整消费后，输入迭代器会移动到窗口的停止边界附近，而不是只移动输出条数。

先把惰性窗口想清楚

islice(iterable, start, stop, step) 的返回值仍然是迭代器。它不会为了找到第 2、4、6 个元素而把输入先复制成一个完整列表，而是在读到需要的位置时才产出结果。对于一次性输入，这个差别比写法本身更重要。

下面的例子用生成器模拟持续到来的记录。代码只取位置 2 到 7 中每隔一个位置的元素：

from itertools import islice

# 生成器代表逐条到达的记录，不先创建完整列表
records = (f"row-{index}" for index in range(10))

# 从位置 2 开始，遇到位置 8 停止，每隔一个位置取一项
window = islice(records, 2, 8, 2)

# 只有消费 window 时，输入生成器才会按需向前读取
print(list(window))

这里的 start=2、stop=8、step=2 对应位置 2、4、6。stop 是开区间，位置 8 不会出现在结果里；step 大于 1 只影响哪些元素被产出，不会让输入跳过到一个完全不同的停止边界。

用 start、stop、step 表达窗口

参数可以写成两种常见形式：islice(source, stop) 表示从 0 取到 stop 之前，islice(source, start, stop[, step]) 表示从 start 开始按步长取值。省略 step 时，它等于 1。

例如做一个“跳过头部，再取一页”的函数时，不需要先把整个输入收集起来：

from itertools import islice

def read_page(records, offset, size):
    # 用 stop 明确本页上界，避免把无界输入一直读下去
    stop = offset + size
    # 返回迭代器，调用方可以按需消费这一页
    return islice(records, offset, stop)

# 这里的 records 可以是文件行生成器或其他一次性迭代器
page = read_page((f"log-{i}" for i in range(100)), 20, 5)
for line in page:
    # 逐条处理，避免为了分页把全部记录放进内存
    print(line)

如果希望从某个位置一直读到输入结束，可以把 stop 写成 None，例如 islice(source, 20, None)。这适合“跳过前 20 条，剩余全部交给后续处理”的场景，但要确认输入最终会结束，否则消费循环也会持续下去。

按数据源选择 islice 或序列切片

我实际做选择时只看一个问题：手里的是可以反复索引的序列，还是只能向前读的迭代器？列表切片支持负索引，并且表达式短；但它会创建新的列表。文件对象、生成器和分页流没有可靠的负索引，也不适合为了切片先物化。

所以，islice 并不是“列表切片的低配替代品”。它的价值在于保留迭代器的输入形态，让窗口边界和消费策略一起被写出来。

处理负值与消费边界

官方文档明确规定，start、stop 和 step 不接受负值，step 还必须大于 0。把列表的 items[-5:] 直接改写成 islice(source, -5, None) 会得到 ValueError，因为迭代器通常不知道“最后五项”在哪里。若确实需要尾部窗口，应让上游提供有限序列，或设计一个单独的环形缓冲区，而不是假设 islice 能倒着读取。

还有一个容易忽略的点：当输入本身是迭代器并且把 islice 完整消费掉时，输入会前进到 max(start, stop) 对应的消费位置，即使 step 很大，也不代表只消耗了结果数量那么少的元素。多个窗口连续读取同一个生成器时，最好把每个窗口的边界写清楚，并把“消费后不能回退”当成接口约束。

from itertools import islice

source = iter(range(10))

# 只保留位置 1、4、7，但完整消费仍会读过 stop 之前的输入
selected = list(islice(source, 1, 8, 3))

# 继续使用同一个迭代器时，它已经不在起点；这里读取的是后续位置
remaining = list(source)
print(selected, remaining)

这段代码的重点不是输出长什么样，而是提醒你：islice 负责定义一个正向读取窗口，不会给一次性迭代器增加随机访问能力。

常见问题

islice 会马上读取整个生成器吗？不会。它返回惰性迭代器，只有消费结果时才继续读取输入；但为了跳过 start 或到达 stop，输入仍可能被前进到相应位置。

为什么不能用负数取最后几项？因为一般迭代器没有长度和反向读取能力。需要尾部窗口时，应改用有限序列或显式维护缓冲区。

把 islice 看成“给一次性数据源加一个正向窗口”就不容易混淆：序列切片解决索引，islice 解决惰性消费；先判断数据源，再决定参数写法。