如何修改Python代码以仅获取列表中指定节点的下一个相邻节点

核心修改逻辑 #

你现有代码用while循环从目标节点的下一个索引开始遍历到列表末尾，所以会输出所有后续元素，直接取消循环，取索引为目标节点索引+1的单个元素即可，同时补充边界判断避免目标节点是列表最后一个元素时出现下标越界报错。

完整可运行示例代码 #

list1 = [5,9,2,4,3]
target_node = 9 # 你指定的目标节点
# 获取目标节点的下标
pos = list1.index(target_node)
# 边界判断：目标节点不是最后一个元素时才取下个节点
if pos < len(list1) - 1:
    print(list1[pos + 1])
else:
    print("指定节点已是列表最后一个元素，无下一个节点")

补充说明 #

如果你的列表中存在重复值，list.index()只会返回第一个匹配元素的下标，如果你需要匹配所有重复元素的下一个节点，可以自行增加遍历逻辑适配。

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