42_DSLR/histogram.py at main · madvid/42_DSLR · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
from pandas import read_csv
import matplotlib.pyplot as plt
import seaborn as sns
import sys
from constants import expected_col, RED, END


if __name__ == "__main__":
	df = read_csv("datasets/dataset_train.csv")
	name_cols = df.columns.values

	# --- Basic verification on the dataset -- #
	if not (all([name in expected_col for name in name_cols]) \
		and ([col in name_cols for col in expected_col])):
		print(RED + "Dataset does not contain exactly the expected columns." + END)
		sys.exit()

	if df.shape[1] != 19:
		print(RED + "Dataset is expected to have 19 columns." + END)
		sys.exit()

	if df.shape[0] == 0:
		print(RED + "Dataset seems to be empty." + END)
		sys.exit()

	# -- Minor formatting of the datafram -- #
	df["Birthday"] = df["Birthday"].astype("datetime64")

	for c in ["Index", "Hogwarts House", "First Name", "Last Name", "Best Hand"]:
		try:
			expected_col.remove(c)
		except:
			continue

	if not all([typ in ["float64", "int64", "datetime64[ns]"] for typ  in df[expected_col].dtypes]):
		print(RED + "At least one column is not of excpected dtype." + END)
		sys.exit()

	# --- Plotting part -- #
	c_palet = ["goldenrod", "green", "red", "dodgerblue"]
	fig, axes = plt.subplots(3, 5, figsize=(20, 12))

	sns.histplot(data=df, x="Birthday", hue="Hogwarts House", stat="probability", binwidth=90, palette=c_palet, multiple="dodge", legend=None, ax=axes[0,0], common_norm=False)
	sns.histplot(data=df, x="Best Hand", hue="Hogwarts House", stat="probability", palette=c_palet, multiple="dodge", legend=None, ax=axes[0,1], common_norm=False)
	sns.histplot(data=df, x="Arithmancy", hue="Hogwarts House", stat="probability", palette=c_palet, multiple="dodge", legend=None, ax=axes[0,2], common_norm=False)
	sns.histplot(data=df, x="Astronomy", hue="Hogwarts House", stat="probability", palette=c_palet, multiple="dodge", legend=None, ax=axes[0,3], common_norm=False)
	sns.histplot(data=df, x="Herbology", hue="Hogwarts House", stat="probability", palette=c_palet, multiple="dodge", legend=None, ax=axes[0,4], common_norm=False)
	sns.histplot(data=df, x="Defense Against the Dark Arts", hue="Hogwarts House", stat="probability", palette=c_palet, multiple="dodge", legend=None, ax=axes[1,0], common_norm=False)
	sns.histplot(data=df, x="Divination", hue="Hogwarts House", stat="probability", palette=c_palet, multiple="dodge", legend=None, ax=axes[1,1], common_norm=False)
	sns.histplot(data=df, x="Muggle Studies", hue="Hogwarts House", stat="probability", palette=c_palet, multiple="dodge", legend=None, ax=axes[1,2], common_norm=False)
	sns.histplot(data=df, x="Ancient Runes", hue="Hogwarts House", stat="probability", palette=c_palet, multiple="dodge", legend=None, ax=axes[1,3], common_norm=False)
	sns.histplot(data=df, x="History of Magic", hue="Hogwarts House", stat="probability", palette=c_palet, multiple="dodge", legend=None, ax=axes[1,4], common_norm=False)
	sns.histplot(data=df, x="Transfiguration", hue="Hogwarts House", stat="probability", palette=c_palet, multiple="dodge", legend=None, ax=axes[2,0], common_norm=False)
	sns.histplot(data=df, x="Potions", hue="Hogwarts House", stat="probability", palette=c_palet, multiple="dodge", legend=None, ax=axes[2,1], common_norm=False)
	sns.histplot(data=df, x="Care of Magical Creatures", hue="Hogwarts House", stat="probability", palette=c_palet, multiple="dodge", legend=None, ax=axes[2,2], common_norm=False)
	sns.histplot(data=df, x="Charms", hue="Hogwarts House", stat="probability", palette=c_palet, multiple="dodge", legend=None, ax=axes[2,3], common_norm=False)
	sns.histplot(data=df, x="Flying", hue="Hogwarts House", stat="probability", palette=c_palet, multiple="dodge", legend=None, ax=axes[2,4], common_norm=False)

	axes[0,1].set_ylabel("")
	axes[0,2].set_ylabel("")
	axes[0,3].set_ylabel("")
	axes[0,4].set_ylabel("")
	axes[1,1].set_ylabel("")
	axes[1,2].set_ylabel("")
	axes[1,3].set_ylabel("")
	axes[1,4].set_ylabel("")
	axes[2,1].set_ylabel("")
	axes[2,2].set_ylabel("")
	axes[2,3].set_ylabel("")
	axes[2,4].set_ylabel("")
	axes[2,2].legend(loc=5, labels=["Hufflepuff", "Gryffindor", "Slytherin", "Ravenclaw"], bbox_to_anchor=(4.03,1.5), fontsize="large")
	plt.show()