Python循环中实现可变滑动窗口的最优方案（处理剩余元素）

我懂你现在的困扰——原来的循环逻辑只处理那些能完整容纳window_size的起始位置，导致末尾剩下的元素（不管是凑不够一个完整窗口，或者像你示例里那样还有一个完整窗口没被覆盖）都被漏掉了。这里有两种通用的解决方案，适配任何window_size、step和tokens长度：

方案1：确保所有元素都被覆盖（允许窗口长度小于window_size） #

如果你的需求是每个元素至少被一个窗口处理到，哪怕最后一个窗口的长度不足window_size，可以用这个循环逻辑：

n = len(tokens)
i = 0
while i < n:
    # 取当前窗口的结束位置，最多到tokens末尾
    end = min(i + window_size, n)
    # 处理当前窗口
    doc2vec.model.infer_vector(tokens[i:end])
    # 如果已经处理到末尾，直接退出循环
    if end == n:
        break
    # 否则按步长前进
    i += step

拿你的示例测试：tokens长度98，window_size=10，step=5，这个循环会处理0:10、5:15...85:95，然后i变成90，处理90:98（最后8个元素），完美覆盖所有元素。

方案2：强制处理末尾的完整窗口（匹配你的示例需求） #

如果你的需求是末尾必须处理一个完整的window_size窗口（像你示例里的88:98），可以先处理所有按步长的完整窗口，再检查是否遗漏了末尾的完整窗口：

n = len(tokens)
i = 0
# 处理所有按步长能取到的完整窗口
while i <= n - window_size:
    doc2vec.model.infer_vector(tokens[i:i+window_size])
    i += step
# 检查是否还有末尾的完整窗口没被处理
if i > n - window_size:
    # 取最后一个完整窗口（从n-window_size到末尾）
    doc2vec.model.infer_vector(tokens[n - window_size:])

这个逻辑在你的示例里，循环会处理到i=85（对应窗口85:95），然后i+step=90，大于n-window_size=88，所以触发末尾处理，处理88:98，刚好符合你想要的结果。

为什么原来的代码会遗漏？ #

你原来的循环终止条件len(tokens)-(window_size+1)是错误的——它让循环提前终止了一个位置，导致最后一个可能的完整起始位置（比如示例里的88）根本没被纳入循环范围。就算修正这个终止条件，步长的存在还是可能让末尾的完整窗口被跳过（比如示例里步长5，85之后是90，超过了88），所以上面的两种方案能更稳妥地解决问题。

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