Boltzmann Partition Function Python Syntax