from random import randint
def ehrenfest(n,k):
    L=[1]*k #On met les k boules dans l'urne 1
    for i in range(n):
        choix=randint(0,k-1) #On choisit une boule au hasard
        if L[choix]==1:
            L[choix]=2 #Si elle est dans l'urne 1, on la met dans la 2
        else:
            L[choix]=1 #Si elle est dans l'urne 2, on la met dans la 1
    return L