Implemented stub that plots the dataset

1 files changed, 31 insertions, 0 deletions

diff --git a/plot.py b/plot.py
new file mode 100644
index 0000000..1669c5a
--- /dev/null
+++ b/plot.py
@@ -0,0 +1,31 @@
+import csv
+import os
+import re
+
+import matplotlib.pyplot as plot
+
+
+def load(directory):
+	def _read(iterable):
+		for x, y in iterable:
+			yield float(x), float(y)
+
+	def _load(filename):
+		with open(filename) as fd:
+			reader = csv.reader(fd)
+			return list(_read(reader))
+
+	def _files(directory):
+		for file in os.listdir(directory):
+			match = re.match(r"SWI_(-?\d+)\.csv", file)
+			if match:
+				yield int(match.group(1)), os.path.join(directory, file)
+
+	return [(x, _load(y)) for x, y in _files(directory)]
+
+
+swis = load("dataset")
+
+for index, data in swis:
+	plot.plot([x[0] for x in data], [x[1] for x in data], label=index)
+plot.show()