Перестановкой элементов квадратной вещественной матрицы добиться того, чтобы её максимальные элементы встали в диагональ

from random import *
from math import *
 
n=int(input("Введите размер кв. матрицы (N): "))
A=[] # Пустой список
B=[]
maximum=0
num=0
lmax=0
mmax=1
for i in range(n):
    A.append([0]*n) # Вложенный список
                    # инициированный 0
    B.append([3]*n) # Вложенный список
                    # инициированный 3
print() # Просто "Enter"
for Row in range(n): # По строками
    for Col in range(n): # по столбцам
        A[Row][Col]=randint(-99,10)
 
for i in range(n):
    flag = True
    num = num + 1
    for j in range(n):
        if (A[i][j]>0):
            flag=False
    if flag == True:
        print("Номер первой строки : " , num)
        break
    
print()
 
for i in range(n): # Вывод результата
    for j in range(n):
            print("{0:02}".format(A[i][j]), end=" ")
    print()

from random import *
from math import *
from heapq import nlargest
 
n=int(input("Введите размер квадратной  матрицы (N): "))
A=[] 
B=[]
maximum=0
num=0
lmax=0
mmax=1
for i in range(n):
    A.append([0]*n) 
    B.append([3]*n)
    
print()
print("Введённая матрица") 
for Row in range(n): 
    for Col in range(n): # постолбцам
        A[Row][Col]=randint(-99,10)       
for i in range(n): # Выводрезультата
    for j in range(n):
            print("{0:02}".format(A[i][j]), end=" ")
    print()
print("Изменённая матрица")
 
matrix = A
maxima = nlargest(n, ((matrix[i][j], i, j,) for i in range(n) for j in range(n)))
for (_, i, j), k in zip(maxima, range(n)):
    matrix[k][k], matrix[i][j] = matrix[i][j], matrix[k][k]
for row in matrix:
     print(*row)
 
for i in range(n):
    num += 1
    flag = True
    for j in range(n):
        if (A[i][j]>0):
            flag=False
    if flag == True:
        print(" Первая строка , которая не содержит ни одного положительного элемента : ",num)
        break