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

🚧 Adds (non-working) boxplot

Browse Source
This commit is contained in:
Daniel Svitan
2024-12-21 18:42:14 +01:00
parent 2b0829c78e
commit 9b90fb2db3
3 changed files with 51 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
analyze_sex.py
View File

@@ -1,6 +1,6 @@
import numpy as np
from analyze import analyze
from analyze import analyze, plot_box
dataset = np.load("clean.npy")
print(f"dataset shape: {dataset.shape}; analyzing column 1 (sex)")
@@ -11,13 +11,19 @@ print("")
def analyze_sex(name: str, col: np.ndarray):
sex_col = dataset[:, 1]
analyze(name, [
data = [
col[sex_col == 0],
col[sex_col == 1]
])
]
F, p = analyze(name, data)
return data, F, p
analyze_sex("gpa", dataset[:, 2])
analyze_sex("math", dataset[:, 3])
analyze_sex("slovak", dataset[:, 4])
analyze_sex("english", dataset[:, 5])
data_gpa, F_gpa, p_gpa = analyze_sex("gpa", dataset[:, 2])
data_math, F_math, p_math = analyze_sex("math", dataset[:, 3])
data_slovak, F_slovak, p_slovak = analyze_sex("slovak", dataset[:, 4])
data_english, F_english, p_english = analyze_sex("english", dataset[:, 5])
plot_box([data_gpa, data_math, data_slovak, data_english], ["Female", "Male"],
[F_gpa, F_math, F_slovak, F_english], [p_gpa, p_math, p_slovak, p_english],
"Pohlavie", ["Priemer", "Matematika", "Slovenčina", "Angličtina"])