plot

plot/main.py

@@ -0,0 +1,47 @@
+import os
+import matplotlib.pyplot as plt
+import numpy as np
+import pandas as pd
+
+"""
+data = pd.read_csv("data.csv", delimiter=';').to_numpy()
+
+tick = data[:, 0]
+
+rb = data[:, 4]
+
+"""
+nb_files = os.listdir('export/')
+
+size = len(nb_files)
+
+# main array
+averages = np.empty((size, 2))
+nb = 0
+for i in nb_files:
+    data = pd.read_csv("export" + os.sep + i, delimiter=';').to_numpy()
+    rb = data[:, 4]
+
+    total = 0.0
+    for x in rb:
+        total = total + x
+    average = total / len(rb)
+    nb_users = i.split(".")[0]
+    averages[nb, 0] = int(nb_users)
+    averages[nb, 1] = average
+    nb += 1
+
+# Data for plotting
+averages.sort(axis=0)
+x = averages[:, 0]
+y = averages[:, 1]
+
+print(averages)
+fig, ax = plt.subplots()
+ax.scatter(x, y)
+
+ax.set(xlabel='users', ylabel='ressources (RB)', title='MaxSNR')
+ax.grid()
+
+# fig.savefig("test.png")
+plt.show()