Py - печать данных массива с использованием ввода

Question

Я занимаюсь информатикой в ​​прошлом (NEA), и у меня есть проблема, когда у меня есть данные, хранящиеся в многомерном массиве, и я спрашиваю ввод от пользователя, где ожидаемый ввод уже находится в массиве, и я хотите, чтобы программа распечатала массив, в котором хранятся входные данные.

# Array containing the ID, last name, year in, membership, nights booked, points.
array = [["San12318", "Sanchez", 2018, "Silver", 1, 2500],
        ["Gat32119", "Gatignol", 2019, "Silver", 3, 7500]]

# Asking to the user to enter the ID
inUser = input("Please enter the user ID: ")

И здесь мне нужна помощь, если введенный идентификатор "San12318", как я могу заставить программу распечатать массив, в котором он хранится?

Answer 1

Как насчет цикла for, который проверяет значение по 0-му индексу для каждой записи данных в списке, то есть значение идентификатора:

def main():
  records = [["San12318", "Sanchez",  2018, "Silver", 1, 2500],
             ["Gat32119", "Gatignol", 2019, "Silver", 3, 7500]]
  input_user_id = input("Please enter a user ID: ")
  print(find_user_id(records, input_user_id.title()))

def find_user_id(records, user_id):
  for record in records:
    if record[0] == user_id:
      return f"Found associated record: {record}"
  return f"Error no record was found for the input user ID: {user_id}"

if __name__ == "__main__":
  main()

Пример использования 1:

Please enter a user ID: san12318
Found associated record: ['San12318', 'Sanchez', 2018, 'Silver', 1, 2500]

Пример использования 2:

Please enter a user ID: Gat42119
Error no record was found for the input user ID: Gat42119