如何在未知索引规则的DataFrame中获取指定索引的下一个紧邻索引？

针对10万行以上的大数据集，且索引类型（数字/字符串）、规则不受控的场景，推荐用以下高效方法实现：

核心思路 #

利用Pandas内置的索引位置查找方法，直接定位输入索引的位置，再取后续位置的索引——这种方式基于哈希表查找，时间复杂度O(1)，完全适配大数据量场景，避免低效遍历。

实现步骤与代码 #

1. 通过df.index.get_loc(input_idx)获取输入索引在索引序列中的位置
2. 判断该位置是否为最后一个索引的位置，若是则返回None（无下一个索引）
3. 若不是最后一位，直接取df.index[pos + 1]得到结果

示例代码

import pandas as pd

# 构造示例DataFrame
data = {'col 1': ['aaa', 'bbb', 'ccc']}
df = pd.DataFrame(data, index=[2342, 2822, 3452])

def get_next_index(df, input_idx):
    try:
        pos = df.index.get_loc(input_idx)
        # 检查是否为最后一个索引
        if pos == len(df.index) - 1:
            return None
        return df.index[pos + 1]
    except KeyError:
        # 处理输入索引不存在的情况
        return None

# 测试
print(get_next_index(df, 2342))  # 输出: 2822
print(get_next_index(df, 3452))  # 输出: None（无后续索引）
print(get_next_index(df, 9999))  # 输出: None（输入索引不存在）

补充说明 #

• 该方法支持数字、字符串等任意可哈希类型的唯一索引，适配绝大多数业务场景
• 若存在重复索引，get_loc会返回布尔数组，此时需要额外处理（比如取第一个匹配位置的下一个索引），但通常业务中索引都是唯一的
• 异常处理部分可根据需求调整，比如抛出提示而非返回None

内容的提问来源于stack exchange，提问作者Hewarathna Ashen Iranga