diff --git a/py_main.py b/py_main.py index 98dea54..0c952ec 100644 --- a/py_main.py +++ b/py_main.py @@ -34,8 +34,8 @@ def kPlusProches(echantillon, donnees, k): """ à reprendre : CHAOS """ voisins = [] for i in range(len(donnees)): - date, conso, t_moy = donnees[i] - d = distance(echantillon, (date, conso, t_moy)) + date, t_moy, t_ref = donnees[i] + d = distance(echantillon, (date, t_moy, t_ref)) voisins.append((d, i)) voisins = sorted(voisins) return [voisins[i][1] for i in range(k)] @@ -51,9 +51,12 @@ def numeroJour(date): mois = (0,31,59,90,120,151,181,212,243,273,304,334,365) return mois[m-1] + j -def PuissanceMoyenne(lst): +def PuissanceMoyenne(consos, i_voisins): """renvoie la puissance moyenne, prend en entrée une liste triée en fonction de la distance""" - + res = 0 + for el in i_voisins: + res += consos[i_voisins] + return res/(len(i_voisins)) def triAvecIndices(lst): @@ -70,4 +73,5 @@ for date in dates: num_jour.append(numeroJour(date)) coords = list(zip(num_jour, t_moys, t_refs)) print(coords) -print(kPlusProches((80, 14, 9), coords, 4)) \ No newline at end of file +print(kPlusProches((80, 14, 9), coords, 4)) +print(PuissanceMoyenne(consos, kPlusProches((80, 14, 9), coords, 4)))