Boltzmann Partition Function Python