Entropy Function Python