Iterador en Python - generar partición

Puedes crear una lista x, cuando recibas un valor, escoge un número entero que vaya desde cero hasta x-1 y pon ese valor en una nueva lista, quedaría así:

def random_partition(k, iterable):
results = [[] for i in range(k)]
for value in iterable:
x = random.randrange(k) 
results[x].append(value)
return results