Python - Pivoteo

Buenas gente!! tengo una duda muy grande! necesito definir una funcion que me permita hacer el pivoteo de una matriz ampliada, para luego a hace matriz aplicarle el metodo de descomposición LU, lo que necesito es que se vaya fijando en la diagonal principal de la matriz y si hay algun coeficiente igual a cero que pivoteee toda la fila por otra que ene ese lugar tenga un coeficiente distinto de cero!
desde ya muchas gracais por cualqueir ayudaa!! y esto me esta matando jeejje