python – ¿Como puedo mejorar este codigo para que no me salga error?

Buenas noches estimados compañeros, tengo un problema, estoy tratando de resolver ecuaciones en matrices por el método de Gauss-Jordan en python, el cual transforma en matriz identidad. El problema es que al compilar el algoritmo me sale como respuesta ‘nan’ en vez de números. Les muestro el codigo y por favor espero que me puedan ayudar. Gracias!

    def gaussjordan1(a,b):
       n=len(b)
       c=np.concatenate((a,b),axis=1)
       for e in range(n):
           c(e,e:)=c(e,e:)/c(e,e)
           for i in range(n):
               if i!=e:
                   c(i,e:)=c(i,e:)-c(i,e)*c(e,e:)
       x=c(:,n)
       return x
   #Resolucion de ejemplo
   from numpy import*
   from gaussjordan1 import*
   a=array(((1,1,-1),(3,3,1),(1,0,1)),float)
   b=array(((2),(2),(0)),float)
   print (a)
   print("======================================")
   print(b)
   print("======================================")
   x=gaussjordan1(a,b)
   print("000000000000000000000000000000000000000000")
   print(x)
   print("000000000000000000000000000000000000000000")