如何为Python列表实现带预热阶段的滑动窗口迭代？

要实现包含预热阶段的滑动窗口遍历，你可以通过以下几种高效方式扩展代码，适配任意大小的列表和窗口尺寸：

方式一：拆分两个循环（逻辑清晰） #

先处理预热阶段的短窗口，再处理固定大小的滑动窗口，代码直观易懂：

seq = [0, 1, 2, 3, 4, 5]
window_size = 3

# 预热阶段：生成长度从1到window_size-1的前缀子列表
for length in range(1, window_size):
    print(seq[:length])

# 固定滑动窗口阶段：生成标准的固定大小窗口
for i in range(len(seq) - window_size + 1):
    print(seq[i:i+window_size])

方式二：合并为单个循环（代码紧凑） #

通过判断当前迭代的位置，动态调整窗口的起始索引，用一个循环完成所有窗口的生成：

seq = [0, 1, 2, 3, 4, 5]
window_size = 3

for end_idx in range(len(seq)):
    # 确定窗口起始位置：预热阶段从0开始，固定窗口阶段从end_idx - window_size +1开始
    start_idx = 0 if end_idx < window_size - 1 else end_idx - window_size + 1
    # 切片生成窗口（左闭右开，所以结束位置是end_idx +1）
    print(seq[start_idx:end_idx+1])

方式三：生成器实现（内存友好） #

如果处理超大列表，推荐用生成器延迟生成窗口，避免一次性占用过多内存：

def sliding_window_with_burnin(seq, window_size):
    # 生成预热窗口
    for length in range(1, window_size):
        yield seq[:length]
    # 生成固定大小窗口
    for i in range(len(seq) - window_size + 1):
        yield seq[i:i+window_size]

# 使用示例
seq = [0, 1, 2, 3, 4, 5]
window_size = 3
for window in sliding_window_with_burnin(seq, window_size):
    print(window)

以上三种方式的时间复杂度均为O(N*k)（N为列表长度，k为窗口大小），这是生成所有窗口元素的必要开销，实际运行中Python的切片操作经过底层优化，效率足够应对绝大多数场景。

内容的提问来源于stack exchange，提问作者BlackHawk