import numpy as np
import pandas as pd
import matplotlib.pyplot as plt


path = "forecasts.csv"

df = pd.read_csv(path, encoding = "utf8")

gens = {
    "Солнце 50%": 2,
    "Ветер 50%": 2
}
cons = {
    "Больницы 50%": 1,
    "Заводы 50%": 2,
    "Дома 50%": 3
}

total_gen = 0
total_con = 0


array_gen = []
array_con = []

for i in range(len(df)):
    total_gen = 0
    total_con = 0
    for j in gens:
        total_gen += gens[j]*df[j][i]
    for j in cons:
        total_con += cons[j]*df[j][i]
    array_gen.append(total_gen)
    array_con.append(total_con)


fig, ax = plt.subplots()

ax.plot(array_con, '-b', label="Consumed")
ax.plot(array_gen, '-g', label="Generated")
ax.legend()

plt.show()