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

@@ 673-696 (lines=24) @@
670
    return df_profile_merge.loc[:, columns]
671
672
673
def plot_heat_supply(resulting_capacities):
674
675
    from matplotlib import pyplot as plt
676
677
    mv_grids = db.select_geodataframe(
678
        """
679
        SELECT * FROM grid.egon_mv_grid_district
680
        """,
681
        index_col="bus_id",
682
    )
683
684
    for c in ["CHP", "heat_pump"]:
685
        mv_grids[c] = (
686
            resulting_capacities[resulting_capacities.carrier == c]
687
            .set_index("mv_grid_id")
688
            .capacity
689
        )
690
691
        fig, ax = plt.subplots(1, 1)
692
        mv_grids.boundary.plot(linewidth=0.2, ax=ax, color="black")
693
        mv_grids.plot(
694
            ax=ax,
695
            column=c,
696
            cmap="magma_r",
697
            legend=True,
698
            legend_kwds={
699
                "label": f"Installed {c} in MW",