Python如何根据输入值获取列表中的下一个元素

核心逻辑 #

• 先匹配输入值在目标列表中的索引位置
• 若输入值不存在于列表、或是列表最后一个元素，直接返回空字符串
• 其余情况返回索引位置的后一位元素

可直接运行的实现代码 #

elem_list = ['a','b','c','d']

def fetch_next_element(input_val: str) -> str:
    # 输入值不在列表范围内直接返回空
    if input_val not in elem_list:
        return ''
    current_pos = elem_list.index(input_val)
    # 已经是最后一个元素，返回空
    if current_pos == len(elem_list) - 1:
        return ''
    return elem_list[current_pos + 1]

预期用例验证 #

• 传入input_val='a'，返回结果为'b'，符合要求
• 传入input_val='c'，返回结果为'd'，符合要求
• 传入input_val='d'，返回结果为''，符合要求

注意：当前实现适配给出的无重复元素列表场景，如果后续列表存在重复值，elem_list.index(input_val)只会返回第一个匹配项的位置，可根据实际需求调整匹配规则。

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