diff --git a/mnp/evaluation/subselection_output.py b/mnp/evaluation/subselection_output.py index bf379d8df0a03e6ca3e7ff9fcb2bc42da7309d7b..fe2c5a65e8eeafa952f549b99369fa335964cbfe 100644 --- a/mnp/evaluation/subselection_output.py +++ b/mnp/evaluation/subselection_output.py @@ -766,9 +766,7 @@ class LandTypeMap(SubselectionOutput): parameters: MNPParameters, output_path: str = "", ): - super().__init__( - output_path=output_path - ) + super().__init__(output_path=output_path) self.subselection_name = subselection_evaluation.name self.land_types = land_types self.species_codes = subselection_evaluation.species_codes @@ -777,15 +775,19 @@ class LandTypeMap(SubselectionOutput): self.legend = None def create(self): - - selected_land_type_codes = self.suitability_indexes.loc[ - self.suitability_indexes.index.isin(self.species_codes) - & self.suitability_indexes.land_type_code.isin(self.land_types.keys()), - ].dropna(axis=0, how="all").land_type_code.unique() + selected_land_type_codes = ( + self.suitability_indexes.loc[ + self.suitability_indexes.index.isin(self.species_codes) + & self.suitability_indexes.land_type_code.isin(self.land_types.keys()), + ] + .dropna(axis=0, how="all") + .land_type_code.unique() + ) land_type_legend = pd.read_csv( - os.path.join(self.parameters.folders["land_types"], "land_types.csv"), - sep=',', comment='#' + os.path.join(self.parameters.folders["land_types"], "land_types.csv"), + sep=",", + comment="#", ) self.legend = land_type_legend.loc[ @@ -793,7 +795,16 @@ class LandTypeMap(SubselectionOutput): ] def save_qlr(self): - d = self.legend.set_index("land_type_id").land_type_name.sort_values().to_dict() + d = ( + self.legend.assign( + land_type_code_name=self.legend.land_type_code.str.cat( + self.legend.land_type_name, sep=": " + ) + ) + .set_index("land_type_id") + .land_type_code_name.sort_values() + .to_dict() + ) template = LandTypeMapTemplate(land_type_id_to_code_name=d) template.find_replace("*SUBSELECTION_NAME*", self.subselection_name) template.write_to_file(destination=os.path.join(f"{self.output_path}.qlr"))