Streamer272/soc-2024
0
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

⚡ Adds sleep graph

Browse Source
This commit is contained in:
Daniel Svitan 2024-12-21 21:09:00 +01:00
parent 2103716f79
commit 0fceaafba6
2 changed files with 18 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
analyze.py
View File

@ -31,9 +31,10 @@ def plot_violin(data, labels, Fs, ps, title):
if not graph:
return
titles = ["Priemer", "Matematika", "Slovenčina", "Angličtina"]
grade_names = ["Priemer", "Matematika", "Slovenčina", "Angličtina"]
grade_name_labels = ["Priemer známok", "Známka z matematiky", "Známka zo slovenčiny", "Známka z angličtiny"]
fig, axs = plt.subplots(2, 2, sharex=True)
fig, axs = plt.subplots(2, 2)
fig.suptitle(title)
fig.set_size_inches(12, 9)
@ -43,7 +44,9 @@ def plot_violin(data, labels, Fs, ps, title):
step = 1 if index > 0 else 0.5
axs[j, k].violinplot(data[index], showmeans=True)
axs[j, k].set_title(titles[index])
axs[j, k].set_title(grade_names[index])
axs[j, k].set_xlabel(title, fontweight="bold")
axs[j, k].set_ylabel(grade_name_labels[index], fontweight="bold")
axs[j, k].set_xticks(np.arange(1, len(labels) + 1), labels=labels)
axs[j, k].set_yticks(np.arange(1, 5.01, step))

19
analyze_sleep.py
View File

@ -1,6 +1,6 @@
import numpy as np
from analyze import analyze
from analyze import analyze, plot_violin
dataset = np.load("clean.npy")
print(f"dataset shape: {dataset.shape}; analyzing column 10 (sleep)")
@ -12,14 +12,19 @@ print("")
def analyze_ses(name: str, col: np.ndarray):
sex_col = dataset[:, 10]
analyze(name, [
data = [
col[sex_col == 0],
col[sex_col == 1],
col[sex_col == 2]
])
]
F, p = analyze(name, data)
return data, F, p
analyze_ses("gpa", dataset[:, 2])
analyze_ses("math", dataset[:, 3])
analyze_ses("slovak", dataset[:, 4])
analyze_ses("english", dataset[:, 5])
data_gpa, F_gpa, p_gpa = analyze_ses("gpa", dataset[:, 2])
data_math, F_math, p_math = analyze_ses("math", dataset[:, 3])
data_slovak, F_slovak, p_slovak = analyze_ses("slovak", dataset[:, 4])
data_english, F_english, p_english = analyze_ses("english", dataset[:, 5])
plot_violin([data_gpa, data_math, data_slovak, data_english], ["Krátky", "Stredný", "Dlhý"],
[F_gpa, F_math, F_slovak, F_english], [p_gpa, p_math, p_slovak, p_english], "Spánok")