Code Duplication    Length = 21-24 lines in 2 locations

src/egon/data/datasets/heat_supply/district_heating.py 1 location

@@ 462-482 (lines=21) @@
459
    return df
460
461
462
def plot_heat_supply(resulting_capacities):
463
464
    from matplotlib import pyplot as plt
465
466
    district_heating_areas = select_district_heating_areas("eGon2035")
467
468
    for c in ["CHP", "solar_thermal_collector", "geo_thermal", "heat_pump"]:
469
        district_heating_areas[c] = (
470
            resulting_capacities[resulting_capacities.carrier == c]
471
            .set_index("district_heating_id")
472
            .capacity
473
        )
474
475
        fig, ax = plt.subplots(1, 1)
476
        district_heating_areas.boundary.plot(
477
            linewidth=0.2, ax=ax, color="black"
478
        )
479
        district_heating_areas.plot(
480
            ax=ax,
481
            column=c,
482
            cmap="magma_r",
483
            legend=True,
484
            legend_kwds={
485
                "label": f"Installed {c} in MW",

src/egon/data/datasets/heat_supply/individual_heating.py 1 location

@@ 695-718 (lines=24) @@
692
    return df_profile_merge.loc[:, columns]
693
694
695
def plot_heat_supply(resulting_capacities):
696
697
    from matplotlib import pyplot as plt
698
699
    mv_grids = db.select_geodataframe(
700
        """
701
        SELECT * FROM grid.egon_mv_grid_district
702
        """,
703
        index_col="bus_id",
704
    )
705
706
    for c in ["CHP", "heat_pump"]:
707
        mv_grids[c] = (
708
            resulting_capacities[resulting_capacities.carrier == c]
709
            .set_index("mv_grid_id")
710
            .capacity
711
        )
712
713
        fig, ax = plt.subplots(1, 1)
714
        mv_grids.boundary.plot(linewidth=0.2, ax=ax, color="black")
715
        mv_grids.plot(
716
            ax=ax,
717
            column=c,
718
            cmap="magma_r",
719
            legend=True,
720
            legend_kwds={
721
                "label": f"Installed {c} in MW",