Foros Python

Versión completa: problema con función Python?
Actualmente estas viendo una versión simplificada de nuestro contenido. Ver la versión completa con el formato correcto.
me piden: Realizar un módulo/funcion que permita modificar la cantidad de pasajeros que subieron en una estación y una formación solicitados por teclado dentro del subprograma.

en mi código se modifican todas los pasajeros y formaciones de las estaciones no solo de una estación por esos me gustaría que me ayuden a encontrar el error
este es el código la función se llama modificar:



Código:
from pyrecord import Record
import numpy as np
import random

Estaciones = Record.create_type("Estaciones","estacion","formacion","pasajeros","horario_establecido","horario_real",estacion = 0,formacion = 0,pasajeros = 0,horario_establecido = 0,horario_real=0)

def imprimir_vector(vector, elementos):
    print("estacion","|","formacion","|","pasajeros","|","horario_establecido","|","horario_real")
    print('     ----------------------------------------------------------------------------')
    for i in range(elementos):
        print(vector[i].estacion,vector[i].formacion,vector[i].pasajeros,
        vector[i].horario_establecido,vector[i].horario_real)


def cargar_datos(vector,cantidad):
    registro = 0
    estacion = 100
    for i in range(cantidad):
        vector[registro] = Estaciones()
        estacion += 1
        vector[registro].estacion = estacion
        vector[registro].formacion = random.randint(1,360)
        vector[registro].pasajeros = random.randint(1,1000)
        vector[registro].horario_establecido = random.randint(8,18)
        vector[registro].horario_real = random.randint(9,23)
        registro += 1
    return registro

def modificar (Vector, cant_real, estacion):
    print('modificar pasajero y formacion')
    Indice = cant_real
    cant_real += 1
    Vector[Indice] = Estaciones
    Vector[Indice].pasajeros = int(input("Ingrese cantidad pasajeros: "))
    Vector[Indice].formacion = int(input("Ingrese formacion: "))
    return Indice





def main():

    max_elementos = 1000

    Subterraneos = np.empty([max_elementos,], dtype=Estaciones)

    cant_real = cargar_datos(Subterraneos,50)


    imprimir_vector(Subterraneos,cant_real)

    N_estacion = int(input('Indique el numero de estacion: '))
    cant_reals = modificar(Subterraneos, cant_real, N_estacion)
    imprimir_vector(Subterraneos,cant_reals)

if __name__ == "__main__":
    main()