Skip to content

Commit a3bffa8

Browse files
measrainseykhaeru
authored andcommitted
Fix median calculation in WEO data
1 parent 10f1ffa commit a3bffa8

File tree

1 file changed

+2
-2
lines changed
  • message_ix_models/tools/costs

1 file changed

+2
-2
lines changed

message_ix_models/tools/costs/weo.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -172,15 +172,15 @@ def get_weo_data() -> pd.DataFrame:
172172

173173
# Calculate median values for each technology
174174
df_median = (
175-
all_cost_df.groupby(["weo_technology"])
175+
all_cost_df.groupby(["weo_technology", "cost_type"])
176176
.agg(median_value=("value", "median"))
177177
.reset_index()
178178
)
179179

180180
# Merge full dataframe with median dataframe
181181
# Replace null values with median values
182182
df_merged = (
183-
all_cost_df.merge(df_median, on=["weo_technology"], how="left")
183+
all_cost_df.merge(df_median, on=["weo_technology", "cost_type"], how="left")
184184
.assign(adj_value=lambda x: np.where(x.value.isnull(), x.median_value, x.value))
185185
.drop(columns={"value", "median_value"})
186186
.rename(columns={"adj_value": "value"})

0 commit comments

Comments
 (0)