Pandas筛选行保留索引仍出现越界错误的解决方案咨询

核心问题 #

你遇到的越界错误，根源是混淆了Pandas的位置索引（iloc）和标签索引（loc）：

• 筛选后的selected保留的是原group的标签索引，但你用iloc[selected.index.values]去取行时，iloc要求传入的是连续的位置序号（0,1,2...）。如果原group的索引不是连续整数（比如自定义索引、之前做过行筛选导致索引断裂），就会出现位置序号超出DataFrame行数的越界问题。

修正步骤 #

直接把最后一步的iloc换成loc，loc通过标签索引定位行，和selected的索引完全匹配：

return group.loc[selected.index]

简化整个流程（可选） #

可以把四步合并成更简洁高效的代码，减少中间变量，从根源避免索引混淆：

# 计算指定列中1的占比
ratio = group.iloc[:, a:b+1].sum(axis=1) / (b - a + 1)
# 筛选符合区间的行并返回
return group[(ratio >= lowerInterval) & (ratio <= upperInterval)]

额外注意事项 #

• 如果必须使用iloc，需要先把标签索引转换成位置索引，用group.index.get_indexer(selected.index)获取对应的位置序号，再传入iloc：

  pos_indices = group.index.get_indexer(selected.index)
  return group.iloc[pos_indices]
  

• 提前校验a和b的取值合法性：确保0 <= a <= b < group.shape[1]，避免列选取阶段先出现越界。

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