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

@@ 586-609 (lines=24) @@
583
    return df_profile_merge.loc[:, columns]
584
585
586
def plot_heat_supply(resulting_capacities):
587
588
    from matplotlib import pyplot as plt
589
590
    mv_grids = db.select_geodataframe(
591
        """
592
        SELECT * FROM grid.egon_mv_grid_district
593
        """,
594
        index_col="bus_id",
595
    )
596
597
    for c in ["CHP", "heat_pump"]:
598
        mv_grids[c] = (
599
            resulting_capacities[resulting_capacities.carrier == c]
600
            .set_index("mv_grid_id")
601
            .capacity
602
        )
603
604
        fig, ax = plt.subplots(1, 1)
605
        mv_grids.boundary.plot(linewidth=0.2, ax=ax, color="black")
606
        mv_grids.plot(
607
            ax=ax,
608
            column=c,
609
            cmap="magma_r",
610
            legend=True,
611
            legend_kwds={
612
                "label": f"Installed {c} in MW",