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

⚡ Adds commute and living violin plot

Browse Source
This commit is contained in:
Daniel Svitan 2024-12-21 19:41:53 +01:00
parent 71603ec299
commit 2103716f79
2 changed files with 25 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
analyze_commute.py
View File

@ -1,6 +1,6 @@
import numpy as np import numpy as np
from analyze import analyze from analyze import analyze, plot_violin
dataset = np.load("clean.npy") dataset = np.load("clean.npy")
print(f"dataset shape: {dataset.shape}; analyzing column 9 (commute)") print(f"dataset shape: {dataset.shape}; analyzing column 9 (commute)")
@ -14,16 +14,21 @@ print("")
def analyze_living(name: str, col: np.ndarray): def analyze_living(name: str, col: np.ndarray):
occupation_col = dataset[:, 9] occupation_col = dataset[:, 9]
analyze(name, [ data = [
col[occupation_col == 0], col[occupation_col == 0],
col[occupation_col == 1], col[occupation_col == 1],
col[occupation_col == 2], col[occupation_col == 2],
col[occupation_col == 3], col[occupation_col == 3],
col[occupation_col == 4] col[occupation_col == 4]
]) ]
F, p = analyze(name, data)
return data, F, p
analyze_living("gpa", dataset[:, 2]) data_gpa, F_gpa, p_gpa = analyze_living("gpa", dataset[:, 2])
analyze_living("math", dataset[:, 3]) data_math, F_math, p_math = analyze_living("math", dataset[:, 3])
analyze_living("slovak", dataset[:, 4]) data_slovak, F_slovak, p_slovak = analyze_living("slovak", dataset[:, 4])
analyze_living("english", dataset[:, 5]) data_english, F_english, p_english = analyze_living("english", dataset[:, 5])
plot_violin([data_gpa, data_math, data_slovak, data_english], ["Intrák", "<= 15m", "<= 30m", "<= 1h", "> 1h"],
[F_gpa, F_math, F_slovak, F_english], [p_gpa, p_math, p_slovak, p_english], "Dochádzanie")

20
analyze_living.py
View File

@ -1,6 +1,6 @@
import numpy as np import numpy as np
from analyze import analyze from analyze import analyze, plot_violin
dataset = np.load("clean.npy") dataset = np.load("clean.npy")
print(f"dataset shape: {dataset.shape}; analyzing column 8 (living)") print(f"dataset shape: {dataset.shape}; analyzing column 8 (living)")
@ -14,16 +14,22 @@ print("")
def analyze_living(name: str, col: np.ndarray): def analyze_living(name: str, col: np.ndarray):
occupation_col = dataset[:, 8] occupation_col = dataset[:, 8]
analyze(name, [ data = [
col[occupation_col == 0], col[occupation_col == 0],
col[occupation_col == 1], col[occupation_col == 1],
col[occupation_col == 2], col[occupation_col == 2],
col[occupation_col == 3], col[occupation_col == 3],
col[occupation_col == 4] col[occupation_col == 4]
]) ]
F, p = analyze(name, data)
return data, F, p
analyze_living("gpa", dataset[:, 2]) data_gpa, F_gpa, p_gpa = analyze_living("gpa", dataset[:, 2])
analyze_living("math", dataset[:, 3]) data_math, F_math, p_math = analyze_living("math", dataset[:, 3])
analyze_living("slovak", dataset[:, 4]) data_slovak, F_slovak, p_slovak = analyze_living("slovak", dataset[:, 4])
analyze_living("english", dataset[:, 5]) data_english, F_english, p_english = analyze_living("english", dataset[:, 5])
plot_violin([data_gpa, data_math, data_slovak, data_english],
["Rodina", "Príslušník/čka", "Sám/a | spolu", "Intrák", "Iné"], [F_gpa, F_math, F_slovak, F_english],
[p_gpa, p_math, p_slovak, p_english], "Životná situácia")