Python中从唯一列表获取指定元素下一个相邻元素的最优Python风格实现方法

这是个很实用的日常需求，Python里有不少简洁又贴合风格的实现方式，我来分享几个最优方案：

方案一：用zip打造最Pythonic的写法 #

这种方法完全贴合Python「用迭代器处理序列」的思路，把原列表和它的切片lst[1:]配对，直接生成「当前元素-下一个元素」的组合，代码简洁到一眼就能懂：

def next_element(lst, target):
    for current, next_elem in zip(lst, lst[1:]):
        if current == target:
            return next_elem
    # 处理两种边界情况：目标不在列表里，或者目标是最后一个元素
    return None

唯一要注意的是，lst[1:]会创建一个新的子列表，对于超大型列表来说会占用额外内存，但常规大小的列表完全不用在意这点。

方案二：用enumerate实现高效遍历 #

如果你的列表特别大，不想创建切片副本，那用enumerate遍历一次是更高效的选择——找到目标元素后立刻返回下一个，不用遍历整个列表：

def next_element(lst, target):
    for idx, elem in enumerate(lst):
        if elem == target:
            if idx < len(lst) - 1:
                return lst[idx + 1]
            else:
                return None
    # 目标不在列表中的情况
    return None

这个方法内存友好，还能明确处理「目标是最后一个元素」的边界场景，适合对性能有要求的场景。

方案三：直观的index方法（适合小列表） #

如果你的列表很小，追求最直观的写法，直接用列表的index()方法找到元素位置，再取下一个元素也完全没问题：

def next_element(lst, target):
    try:
        idx = lst.index(target)
        return lst[idx + 1] if idx < len(lst) - 1 else None
    except ValueError:
        # 目标不在列表里时触发异常，这里返回None，也可以根据需求抛自定义异常
        return None

不过要注意，index()方法会遍历整个列表找目标，而且如果元素不存在会抛出ValueError，所以需要用try-except捕获异常。

关于边界场景的补充 #

上面的方案都默认在「目标是最后一个元素」或「目标不存在」时返回None，你可以根据实际需求调整：

• 如果希望在目标不存在时抛出异常，去掉try-except或者主动抛出ValueError
• 如果需要循环取元素（比如最后一个元素的下一个是第一个），可以把返回逻辑改成lst[(idx + 1) % len(lst)]

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