diff --git a/projet_knn_feat_denis_le_goat.py b/projet_knn_feat_denis_le_goat.py index b75657b..7675805 100644 --- a/projet_knn_feat_denis_le_goat.py +++ b/projet_knn_feat_denis_le_goat.py @@ -67,10 +67,10 @@ def puissanceMoyenne(listeTriee, tableau): """Prend une liste triée de tuple avec (distance, indice du tableau initial) et un tableau de donnée et renvoie la moyenne de la puissance electrique des k plus proches voisins""" - PuissanceTotale = 0 + PuissanceTotale = 0.0 for donnee in listeTriee: distance,indice = donnee - PuissanceTotale += tableau[indice][1] + PuissanceTotale += tableau[int(indice)][1] return PuissanceTotale/len(listeTriee) def listeTuples(fichier): """Prend une liste de tuples de type (date,pic journalier consommation, @@ -94,7 +94,7 @@ def main(): listeVoisins = kPlusProches((date,tempMoy,tempRef),(listeTuples(fichier)), nombreK) print(listeVoisins) - + print(puissanceMoyenne(listeVoisins,fichier)) fenetre = tk.Tk() fenetre['bg'] = "lavender"