From a35565eb2ee1cccce726e912cd8a66c8126c0d8a Mon Sep 17 00:00:00 2001 From: Aki Date: Fri, 26 Aug 2022 21:06:41 +0200 Subject: Implemented stub that plots the dataset --- plot.py | 31 +++++++++++++++++++++++++++++++ 1 file changed, 31 insertions(+) create mode 100644 plot.py (limited to 'plot.py') 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() -- cgit v1.1