From 5c695c75e66e738799f42473e3fa44a1e6be6a0c Mon Sep 17 00:00:00 2001 From: Lasse Studion Date: Fri, 27 Jan 2023 09:26:30 +0100 Subject: [PATCH] Now downloads file from skv.se --- dotplot.py | 12 +++++++++--- 1 file changed, 9 insertions(+), 3 deletions(-) diff --git a/dotplot.py b/dotplot.py index 0b1abab..d58f9c7 100644 --- a/dotplot.py +++ b/dotplot.py @@ -1,9 +1,15 @@ from matplotlib import pyplot as plt +import requests import pandas as pd -# Import data and rename relevant column. -file_path = input() -df = pd.read_excel(file_path, header=6, usecols=['Import/export']) +# Get data from file "Förbrukning och tillförsel per timme (i normaltid)" +# Statistics available at https://www.svk.se/om-kraftsystemet/kraftsystemdata/elstatistik/ +url = 'https://www.svk.se/siteassets/1.om-kraftsystemet/kraftsystemdata/statistik/forbrukning-tillforsel-per-timme/n_fot2022-01-12.xls' # Data for 2022 +r = requests.get(url) +open('Förbrukning och tillförsel per timme (i normaltid).xls', 'wb').write(r.content) + +# Put the data in a dataframe. +df = pd.read_excel('Förbrukning och tillförsel per timme (i normaltid).xls', header=6, usecols=['Import/export']) df.rename(columns={'Import/export': 'balance'}, inplace=True) # Create columns.