Update app.py - fix median bug

Change drop to skipna for median method
This commit is contained in:
quandanrepo 2023-08-09 16:37:12 +01:00 committed by GitHub
parent 07af58eb85
commit e146302fb9
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23

View file

@ -86,7 +86,7 @@ HEAT_DEMAND_RESPONSE = "ENERGY_CONSUMPTION_CURRENT"
def make_cleaning_averages(df):
# Define a custom function to calculate the median, excluding missing values
def median_without_missing(group):
return group[AVERAGE_FIXED_FEATURES].dropna().median()
return group[AVERAGE_FIXED_FEATURES].median(skipna=True)
cleaning_averages = df.groupby(
["PROPERTY_TYPE", "BUILT_FORM", "CONSTRUCTION_AGE_BAND", "NUMBER_HABITABLE_ROOMS", "NUMBER_HEATED_ROOMS"],