Partition Function In Python Example