Just a little something I hacked together to make my homework easier by calculating entropy and gini impurity, a principal foundation for the decision tree.
Write the .csv file of your choice on line 71. Specify a target feature on line 72, and a method of calculation (entropy/gini) on line 76. This program as-is will give the first two splits for your decision tree.
Run with
python entro.py