Question on how to store Python variable values

def search_f(): #search function
    search_value = int(input("""
    To search for a punishment name by name 1
    If you want to see the list of names of students, you can choose 2
    If you want to see the list of punishments, please enter 3 "") #What features do you want to use?
    if search_value == 1:
      s_name = input ("Please enter your school year_name")
      print(i[s_name])
    elif search_value == 2:
      print(i.keys())
    elif search_value == 3:
      print(i.values())