diff --git a/.gitignore b/.gitignore index 1a2920d..79a9c70 100644 --- a/.gitignore +++ b/.gitignore @@ -1,3 +1,4 @@ doc/*/*.* !doc/*/*.tex !doc/*/*.bib +*.cpython-311.pyc \ No newline at end of file diff --git a/scripts/reader.py b/scripts/reader.py new file mode 100644 index 0000000..6af7e0f --- /dev/null +++ b/scripts/reader.py @@ -0,0 +1,35 @@ +import csv + +class Reader(): + def __init__(self) -> None: + pass + + def readFile(self, filename) -> dict: + with open(filename, 'r') as file: + data = list(csv.reader(file, delimiter=',')) + + # Delete metadata for now + print("Removing metadata") + for i in range(15): + print(data.pop(0)) + + # {'TIME': [data], 'CH1': [data], 'CH2': [data]} + channels = [] + for i in range(len(data[0])): + channels.append([]) + + for step in data[1:]: + for idx, point in enumerate(step): + channels[idx].append(float(point)) + + result = {} + for idx, label in enumerate(data[0]): + result[label] = channels[idx] + + return result + + +if __name__ == '__main__': + reader = Reader() + data = reader.readFile(r".\data\T0007ALL.csv") + print(data) \ No newline at end of file diff --git a/scripts/reflection.py b/scripts/reflection.py new file mode 100644 index 0000000..78d37a4 --- /dev/null +++ b/scripts/reflection.py @@ -0,0 +1,10 @@ +from matplotlib import pyplot as plt +import reader + +data_reader = reader.Reader() +data = data_reader.readFile(r".\data\T0001CH1.csv") + + + +plt.plot(data['TIME'], data['CH1']) +plt.show() \ No newline at end of file