Partition Function In Python Geeks