Boltzmann Partition Function Python Example