48 lines
855 B
Python
48 lines
855 B
Python
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()
|