Untitled

n = int(input("¿Cuántas casas quieres comprobar?: "))
casas = {}

for c in range(n): # Crea el diccionario que contiene el valor de cada casa
    casa = c + 1

    casas[casa] = {}
    valor = input(f"Inserte la generación de la casa {casa}: ")
    casas[casa]["generacion"] = valor
    valor = input(f"Inserte el consumo de la casa {casa}: ")
    casas[casa]["consumo"] = valor
    valor = input(f"Inserte la necesidad de la casa {casa}: ")
    casas[casa]["necesidad"] = valor
    valor = input(f"Inserte el almacenamiento de la casa {casa}: ")
    casas[casa]["almacenamiento"] = valor
    valor = input(f"Inserte la capacidad de la casa {casa}: ")
    casas[casa]["capacidad"] = valor

# Això ho pots canviar

print("\nIndique la función que quiere usar: ") # Pregunta que funcion se quiere usar
print("1 + Comparación de la generación entre casas ")
print("2 + Comparación del consumo entre casas ")    
print("3 + Comparación de la necesidad entre casas ")    
print("4 + Comparación del almacenamiento entre casas ")    
print("5 + Comparación de la capacidad entre casas ")    

funcion = int(input("Introduzca el número de la función: "))

comparar = input("¿Qué casas desea comparar? Separe el número de las casas con un espacio: ")

a, b = comparar.split(" ") # a = numero de la primera casa, b = numero de la segunda casa

if funcion == 1:
    casaA = casas[a]['generacion'] # agafem el valor de la generació de la casa A
    casaB = casas[b]['generacion'] # agafem el valor de la generació de la casa B

    res = casaA - casaB # A partir d'aqui fes el q vulguis amb cada funció i fes el que hagis de fer
    if res < 0:
        print(f"La casa número {a} necesitará {abs(res)}kWh de la casa número {b} para que la generación sea")
    print("1213")