ALGORITHME KRUSKAL - PRODUIT DE MATRICE

class Arete:
    def __init__(self, initial, final, cout):
        self.sommet_initial=initial
        self.sommet_final=final
        self.cout_arrete=cout
def saisie_arete():
    print("Veuillez saisir une arrête: ")
    si=int(input("Sommet initial: "))
    sf=int(input("Sommet final: "))
    cout=float(input("Cout: "))
    a = Arete (si,sf,cout)
    return a
class Graphe:
    def __init__(self,nbsommets,listearetes):
        self.nbsommets=nbsommets
        self.listearetes=listearetes
def saisie_graphe():
    listearetes=[]
    nbsommets=int(input("Nombre de sommets: "))
    nbaretes=int(input("Nombre d'aretes: "))
    for i in range(0, nbaretes):
        arete =[]
        print("Saisissez l'arete numero ", i+1)
        a=saisie_arete()
        arete.append(a.sommet_initial)
        arete.append(a.sommet_final)
        arete.append(a.cout_arrete)
        listearetes.append(arete)
    g = Graphe (nbsommets,listearetes)
    return g
def adjacence(g):
    mat = [[0 for i in range(g.nbsommets)]for i in range(g.nbsommets)]
    for i in range(len(g.listearetes)):
        mat[g.listearetes[i][0]][g.listearetes[i][1]] = g.listearetes[i][2]
        mat[g.listearetes[i][1]][g.listearetes[i][0]] = g.listearetes[i][2]
    return mat
g = saisie_graphe()
h= saisie_graphe()
print (adjacence(g))
print(adjacence(h))

M1=np.array(adjacence(g))
M2=np.array(adjacence(h))
def produit(M1,M2):
    M3=[0][0]
    if len(M1)==len(M1[0])==len(M2)==len(M2[0]):
        np.dot(M1,M2)
    else:
        print("Erreur de dimensions")
    return M3

def produit(M1, M2):
  M3 = []
  if len(M1[0]) != len(M2):
    print("Erreur de dimension")
    return False
  for i in range(len(M1)):
    ligne = []
    for j in range(len(M2[0])):
      element = 0
      for k in range(len(M1[0])):
        element = element + M1[i][k] * M2[k][j]
      ligne.append(element)
    M3.append(ligne)
  return M3
print("M1*M2 = ",produit(M1,M2))