Generator for integer partitions « Python recipes « ActiveState Code