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

⚡ Adds graphing occupation

Browse Source
This commit is contained in:
Daniel Svitan
2024-12-21 19:34:11 +01:00
parent 7df964a000
commit 71603ec299
4 changed files with 24 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
analyze_occupation.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 7 (occupation)")
@@ -15,17 +15,23 @@ print("")
def analyze_occupation(name: str, col: np.ndarray):
occupation_col = dataset[:, 7]
analyze(name, [
data = [
col[occupation_col == 0],
col[occupation_col == 1],
col[occupation_col == 2],
col[occupation_col == 3],
col[occupation_col == 4],
col[occupation_col == 5]
])
]
F, p = analyze(name, data)
return data, F, p
analyze_occupation("gpa", dataset[:, 2])
analyze_occupation("math", dataset[:, 3])
analyze_occupation("slovak", dataset[:, 4])
analyze_occupation("english", dataset[:, 5])
data_gpa, F_gpa, p_gpa = analyze_occupation("gpa", dataset[:, 2])
data_math, F_math, p_math = analyze_occupation("math", dataset[:, 3])
data_slovak, F_slovak, p_slovak = analyze_occupation("slovak", dataset[:, 4])
data_english, F_english, p_english = analyze_occupation("english", dataset[:, 5])
plot_violin([data_gpa, data_math, data_slovak, data_english],
["Veľa práce", "Málo práce", "Šport", "Hudba", "Iné", "Žiadne"],
[F_gpa, F_math, F_slovak, F_english], [p_gpa, p_math, p_slovak, p_english], "Práca alebo aktivita")