From 7ffbde6c2cdd9c05363f38d14373164c8b01a6a1 Mon Sep 17 00:00:00 2001 From: Daniel Svitan Date: Sun, 15 Dec 2024 12:18:33 +0100 Subject: [PATCH] :zap: Adds commute owanova --- analyze_commute.py | 29 +++++++++++++++++++++++++++++ 1 file changed, 29 insertions(+) create mode 100644 analyze_commute.py diff --git a/analyze_commute.py b/analyze_commute.py new file mode 100644 index 0000000..3a53621 --- /dev/null +++ b/analyze_commute.py @@ -0,0 +1,29 @@ +import numpy as np + +from analyze import analyze + +dataset = np.load("clean.npy") +print(f"dataset shape: {dataset.shape}, analyzing column 9 (commute)") +print("\t0 - dorms") +print("\t1 - <= 15m") +print("\t2 - <= 30m") +print("\t3 - <= 1h") +print("\t4 - > 1h") +print("") + + +def analyze_living(name: str, col: np.ndarray): + occupation_col = dataset[:, 9] + analyze(name, [ + col[occupation_col == 0], + col[occupation_col == 1], + col[occupation_col == 2], + col[occupation_col == 3], + col[occupation_col == 4] + ]) + + +analyze_living("gpa", dataset[:, 2]) +analyze_living("math", dataset[:, 3]) +analyze_living("slovak", dataset[:, 4]) +analyze_living("english", dataset[:, 5])