Pandas应用可迭代掩码：值不满足条件则取下一个符合项

原实现无法得到预期结果的核心问题有三点：

• chain.from_iterable返回的是迭代器对象，直接和整数类型的possible_overload_cycle做相等判断永远为False，匹配分支永远不会触发
• 嵌套chain.from_iterable属于错误用法，整数不是可迭代对象，真遍历这个生成的迭代器会直接抛出类型错误，这层封装完全冗余
• 代码没有实际遍历候选索引序列，也没有实现「当前值不满足条件就自动遍历取下一个有效值」的逻辑

你的需求本质是按步长5生成候选索引队列，优先校验传入值是否匹配首个候选位置，不匹配就遍历队列取第一个符合要求的有效值，修正后代码如下：

import pandas as pd

hyst = pd.DataFrame({"test":[12, 4, 5, 4, 1, 3, 2, 5, 10, 9, 7, 5, 3, 6, 3, 2 ,1, 5, 2]})
possible_overload_cycle = 1

# 直接按步长5生成候选索引列表
candidate_pos = list(range(0, len(hyst)-1, 5))

if candidate_pos[0] == possible_overload_cycle:
    hyst_overload_cycle = possible_overload_cycle
else:
    # 遍历候选序列，取第一个满足阈值条件的值
    hyst_overload_cycle = next(pos for pos in candidate_pos if pos >= possible_overload_cycle)

print(hyst_overload_cycle)

运行代码输出为5，和你的预期结果一致。

如果后续需要把判断条件从「索引匹配」改成「对应行的字段值匹配」，只需要修改next里的判断逻辑即可，比如要找test列值等于3的第一个候选位置，把判断条件写成hyst.loc[pos, 'test'] == 3就行，整体遍历逻辑不需要改动。

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