From a87dd43dca19dd271136f84d53e04be18cb94817 Mon Sep 17 00:00:00 2001
From: Lasse Edfast <8794658+lasseedfast@users.noreply.github.com>
Date: Tue, 24 Jan 2023 10:55:34 +0100
Subject: [PATCH] Create dotplot.py

---
 dotplot.py | 25 +++++++++++++++++++++++++
 1 file changed, 25 insertions(+)
 create mode 100644 dotplot.py

diff --git a/dotplot.py b/dotplot.py
new file mode 100644
index 0000000..5e02a83
--- /dev/null
+++ b/dotplot.py
@@ -0,0 +1,25 @@
+from matplotlib import pyplot as plt
+import pandas as pd
+
+# Import data and rename relevant column.
+file_path = input()
+df = pd.read_excel(file_path, header=6, usecols=['Import/export'])
+df.rename(columns={'Import/export': 'balance'}, inplace=True)
+
+# Create columns.
+df['hour']=df.index
+df['y']=df['hour'].apply(lambda x: int(x/120))
+df['x']=df['hour'].apply(lambda x: int(x%120))
+
+# Set export to orange and import to grey.
+df['color'] = df['balance'].apply(lambda x: 'orange' if x < 0 else 'grey')
+
+# Plot the data.
+ax = df.plot(kind='scatter', x='x', y='y', c='color')
+
+# Hide irrelevant names on axes.
+ax.axes.get_xaxis().set_visible(False)
+ax.axes.get_yaxis().set_visible(False)
+
+# Show plot.
+plt.show()