Python中已知列表指定元素，如何获取该元素对应的下一个列表项

你可以通过先获取目标元素的索引，再取索引+1位置的元素实现需求，完整代码如下：

list_numbers = ['l1.pkl', 'l2.pkl', 'l3.pkl', 'l4.pkl', 'l5.pkl', 'l6.pkl']
element = 'l3.pkl'

try:
    # 获取目标元素的索引
    current_idx = list_numbers.index(element)
    # 避免索引越界，判断当前元素不是最后一项
    if current_idx < len(list_numbers) - 1:
        next_element = list_numbers[current_idx + 1]
        print(next_element)
    else:
        # 可根据自身需求修改最后一项的返回逻辑
        print("当前元素已是列表最后一个，无后续相邻元素")
except ValueError:
    # 捕获目标元素不在列表中的异常
    print("给定元素不存在于列表中")

运行上述代码会直接输出l4.pkl，匹配你要的结果。

特殊场景补充说明 #

• 如果列表中存在多个和目标元素相同的项，list.index()只会返回第一个匹配项的索引，如果你需要返回所有匹配项的下一个相邻元素，可以用遍历实现：

def get_all_next(lst, target):
    result = []
    for i in range(len(lst) - 1):
        if lst[i] == target:
            result.append(lst[i+1])
    return result

# 测试示例
test_lst = ['l3.pkl', 'a.pkl', 'l3.pkl', 'b.pkl', 'c.pkl']
print(get_all_next(test_lst, 'l3.pkl')) # 输出 ['a.pkl', 'b.pkl']

• 你可以根据自己的业务需要修改异常情况、边界情况的返回值，比如没有下一个元素时返回None而非打印提示。

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