from math import *
from random import *
import numpy as np
import matplotlib.pyplot as pp


def nombredeboules(a):
    s=0
    for i in range(len(a)):
        s=s+a[i]
    return s


def transfert(a,b):
    N=len(a)
    i=randint(0,N-1) # on choisit un n° entre 0 et N-1
    if a[i]==1: #on échange la case i des urnes A et B
        a[i]=0
        b[i]=1
    else:
        a[i]=1
        b[i]=0
    return(a,b)


def nbA(N,t):
    a=N*[1] # on mets toutes les boules dans A
    b=N*[0] # B est vide
    for i in range(t):
        transfert(a,b)
    return(a,b)

def TracerNbA(N,t):
    a=N*[1] # on mets toutes les boules dans A
    b=N*[0] # B est vide
    x=t*[0]
    y=t*[0]
    for i in range(t):
        x[i]=i
        y[i]=nombredeboules(a)
        transfert(a,b)


    pp.plot(x,y)
    pp.show()
    #return(x,y)


for k in range(25):
    print(nbA(7,k))