simplify: drop inaccurate columns only if no simpl wildcard is used

This commit is contained in:
martacki 2021-05-25 12:58:23 +02:00
parent b9fdacad4d
commit 1bd3a5a806

View File

@ -361,8 +361,8 @@ if __name__ == "__main__":
if snakemake.wildcards.simpl:
n, cluster_map = cluster(n, int(snakemake.wildcards.simpl))
busmaps.append(cluster_map)
n.buses = n.buses.drop(['substation_lv', 'substation_off', 'under_construction'], axis=1)
else:
n.buses = n.buses.drop(['symbol', 'tags', 'under_construction', 'substation_lv', 'substation_off'], axis=1)
n.export_to_netcdf(snakemake.output.network)
busmap_s = reduce(lambda x, y: x.map(y), busmaps[1:], busmaps[0])