Python - PROGRAMA PYTHON ESTRUCTURAS DE DATOS NUMPY

#En esta primera linea llamamos la libreria numpy

import numpy as np

# Ahora creamos la primer clase llamada escuela con la que ingresaremos los datos de cada estudiante

class escuela():

       def __init__(self):

           self.nombre=" "

           self.edad=0

           self.categoria= " "

# despues de tener creados los atributos de la clase pasamos a crear los metodos

       def Registro(self):

           print("---------Menu registro--------\n")

           self.nombre = input("Por favor ingrse su nombre: ")

           self.edad = int(input("por favor ingrese su edad: "))

           while True: #mediante el while true validaremos el menu para que solo se ingrese una de las opciones solicitadas

                print ("Por favor elija una categoria: ")

                print("1:- Cuerdas")

                print("2:- vientos")

                print("3:- percusion")

                opt= int(input(" "))

                if opt<=0 or opt>3: # si la opcion elegida es diferente al menu este condcinal nos vuelve a presentar el menu

                  print("Ingrese opcion valida")

                if opt ==1:

                   self.categoria=("Cuerdas")#mediante los condicionales IF se guardara segun la opcion elegida en la variable asignada

                   break

                elif opt==2:

                   self.categoria=("Vientos")#mediante los condicionales IF se guardara segun la opcion elegida en la variable asignada

                   break

                elif opt==3:

                   self.categoria=("Percusion")#mediante los condicionales IF se guardara segun la opcion elegida en la variable asignada

                   break

       def Imprimir(self):

           print ("NOMBRE" "-", self.nombre,"-","EDAD" "-", self.edad, "-","CATEGORIA" "-", self.categoria)

# creamos la segunda clase solicitada por la guia de actividades

objeto = np.empty((5,), dtype=np.object_)

class main():#Creamos una clase para el menu principal

    def __init__(self):

       self.opt=0

       self.contar=""

# el metodo llamado principal tomara la opcion elegida por el usuario

    def Principal(self):

        opt=1

        contar=0

        while True:# Al igual que en el menu anterior, el while rue permite validar que la opcion elegida si pertenezca al menu, de lo contrario nos vuelve a mostrar el menu.

            print("--------Menu principal---------\n")

            print("1.- Registrar estudiantes")

            print("2.- Imprimir lista de estudiantes")

            print("3.- Salir")

            opt= int(input("Elija una opcion: "))

            if opt ==1:                #Con la condiciona.l del if anidado recogeremos la informacion digitada por el usuario

                 objeto[contar]=escuela()

                 objeto[contar].Registro()

                 contar=contar+1

            elif opt ==2:# con esta opcion se mostraran en pantalla la cantidad de registros.

                  i=0

                  print("--------- Inscritos--------\n")

                  while i < contar:#Por medio de esta opcion se imprimiran todos los datos que acumulemos en la variable i hasta que esta variable iguale al contar

                      objeto[i].Imprimir()

                      i=i+1

            elif opt ==3:# esta opcion permite salir del programa

                    print("Muchas gracias por utilizar nuestro programa ")

                    break

menu=main()# cramos la instancia para llamar la clase

menu.Principal()# Instanciamos el metodo principal.