[pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci
This commit is contained in:
pre-commit-ci[bot] 2023-07-22 11:26:39 +00:00
parent d36bdaf6bb
commit 615684d124
3 changed files with 57 additions and 1043 deletions

File diff suppressed because one or more lines are too long

View File

@ -7,7 +7,7 @@ PRODUCTION_PLOTS = [
"production_deviation_bar",
"seasonal_operation_area",
]
CROSS_BORDER_PLOTS = ["trade_time_series","cross_border_bar"]
CROSS_BORDER_PLOTS = ["trade_time_series", "cross_border_bar"]
PRICES_PLOTS = ["price_bar", "price_line"]

View File

@ -4,10 +4,10 @@
#
# SPDX-License-Identifier: MIT
import country_converter as coco
import matplotlib.pyplot as plt
import pandas as pd
import pypsa
import country_converter as coco
import seaborn as sns
from _helpers import configure_logging
@ -51,6 +51,7 @@ color_country = {
"SK": "#d3e75a",
}
def sort_one_country(country, df):
indices = [link for link in df.columns if country in link]
df_country = df[indices].copy()
@ -64,8 +65,7 @@ def sort_one_country(country, df):
def cross_border_time_series(countries, data):
fig, ax = plt.subplots(2*len(countries), 1, figsize=(15, 10*len(countries)))
fig, ax = plt.subplots(2 * len(countries), 1, figsize=(15, 10 * len(countries)))
axis = 0
for country in countries:
@ -120,7 +120,7 @@ def cross_border_time_series(countries, data):
axis = axis + 1
for x in range(axis-2,axis):
for x in range(axis - 2, axis):
ax[x].set_ylim([neg_min, pos_max])
fig.savefig(snakemake.output.trade_time_series, bbox_inches="tight")
@ -220,7 +220,9 @@ if __name__ == "__main__":
optimized = pd.concat([optimized_links, optimized_lines], axis=1)
# Drop internal country connection
optimized.drop([c for c in optimized.columns if c[:2] == c[5:]], axis=1, inplace=True)
optimized.drop(
[c for c in optimized.columns if c[:2] == c[5:]], axis=1, inplace=True
)
# align columns name
for c1 in optimized.columns:
@ -230,7 +232,7 @@ if __name__ == "__main__":
optimized = optimized.groupby(lambda x: x, axis=1).sum()
cross_border_bar(countries,[historic, optimized])
cross_border_bar(countries, [historic, optimized])
cross_border_time_series(countries, [historic, optimized])