Python, simulation chaîne de Markov

import numpy as np
import matplotlib.pyplot as plt
#Fonction générant une v.a de loi $P_i $
def generate(i,P):
#taille de la matrice  $P $
[nrow,ncol]=np.shape(P)
#Somme cumulé de chaque ligne de $P $
scP=np.cumsum(P,axis=1)
#tirage de v.a uniforme sur $(0,1) $
u=np.random.uniform (1)
if u<=scP[i,0]:
      return(0)
else:
       for j in range(1,ncol):
              if scP[i,j-1]<u and u<=scP[i,j]:
                   return(j)
                    break 
#fonction simulant les trajectoires de la chaîne de #Markov 
def Simul_Markov(P,n,init):
#vecteur trajectoire X
          X=np.zeros(n,dtype=np.int)
           X[0]=init
           for k in range(1,n):
                  X[k]=generate(X[k-1],P)
            plt.plot(X)
            plt.show()