Gauss Elimination with partial pivot - python code

Question

I am trying to create python code that will do Gauss Elimination with Partial Pivot. Below code works for Gauss Elimination, but I am having trouble getting the Partial Pivot t