Boltzmann Partition Function Python Example Code