From cb4241ff979ee041734ffd339e776568d0f55d79 Mon Sep 17 00:00:00 2001
From: Daniel Svitan <daniel.svitan.team7274dev@gmail.com>
Date: Sun, 15 Dec 2024 12:20:56 +0100
Subject: [PATCH] :zap: Adds sleep owanova

---
 CLEAN.md         |  6 +++---
 analyze_sleep.py | 25 +++++++++++++++++++++++++
 2 files changed, 28 insertions(+), 3 deletions(-)
 create mode 100644 analyze_sleep.py

diff --git a/CLEAN.md b/CLEAN.md
index a56febf..666ba7e 100644
--- a/CLEAN.md
+++ b/CLEAN.md
@@ -66,7 +66,7 @@ The cleaned dataset will have the following structure:
 ### Sleep
 
 ```
-0 - long
-1 - medium
-2 - short
+0 - long sleepers
+1 - medium sleepers
+2 - short sleepers
 ```
diff --git a/analyze_sleep.py b/analyze_sleep.py
new file mode 100644
index 0000000..c9ce9be
--- /dev/null
+++ b/analyze_sleep.py
@@ -0,0 +1,25 @@
+import numpy as np
+
+from analyze import analyze
+
+dataset = np.load("clean.npy")
+print(f"dataset shape: {dataset.shape}, analyzing column 10 (sleep)")
+print("\t0 - long sleepers")
+print("\t1 - medium sleepers")
+print("\t2 - short sleepers")
+print("")
+
+
+def analyze_ses(name: str, col: np.ndarray):
+    sex_col = dataset[:, 10]
+    analyze(name, [
+        col[sex_col == 0],
+        col[sex_col == 1],
+        col[sex_col == 2]
+    ])
+
+
+analyze_ses("gpa", dataset[:, 2])
+analyze_ses("math", dataset[:, 3])
+analyze_ses("slovak", dataset[:, 4])
+analyze_ses("english", dataset[:, 5])