@@ -0,0 +1,802 @@
+from matplotlib import pyplot as plt%0Aimport numpy as np%0A%0A%0Adef plot_line_expansion(edisgo_obj,timesteps):%0A%0A    pypsa_plot = edisgo_obj.to_pypsa(mode='mv',timesteps=timesteps)%0A    bus_index = pypsa_plot.buses.index%0A    pypsa_plot.buses.x.loc%5Bbus_index%5D = edisgo_obj.topology.mv_grid.buses_df.x.loc%5Bbus_index%5D%0A    pypsa_plot.buses.y.loc%5Bbus_index%5D = edisgo_obj.topology.mv_grid.buses_df.y.loc%5Bbus_index%5D%0A    line_colors = edisgo_obj.opf_results.lines.squeeze()%0A    plt.figure(figsize=(12, 8))%0A    ax = plt.gca()%0A    v = np.linspace(min(line_colors), max(line_colors), 101)%0A    psa_plot = pypsa_plot.plot(line_colors=line_colors, line_cmap=plt.cm.cool, ax=ax, geomap=False)%0A    cb = plt.colorbar(psa_plot%5B1%5D)%0A    cb.set_clim(vmin=v%5B0%5D, vmax=v%5B-1%5D)%0A    cb.set_label('Number of installed lines')%0A%0A    return%0A

add plotting of line expansion