无法以角度反应形式从多个单选中检索值

def retrieve_value(option):
    if option == 1:
        return "值1"
    elif option == 2:
        return "值2"
    elif option == 3:
        return "值3"
    else:
        return "无效选项"

value = retrieve_value(2)
print(value)  # 输出：值2

def retrieve_value(option):
    value_dict = {
        1: "值1",
        2: "值2",
        3: "值3"
    }
    return value_dict.get(option, "无效选项")

value = retrieve_value(3)
print(value)  # 输出：值3