calculate entropy of dataset in python