Если не все диагональные элементы матрицы положительны, заменить все простые числа в матрице

from random import randint
 
def prnum(n) :
    prime = [True] * (n+1)
    prime[0] = prime[1] = False
    prime[2] = True
    for j in range(4, n+1, 2):
        prime[j] = False
    for i in range(3, n + 1, 2) :
        if not prime[i]:
            continue
        for j in range(i * i, n+1, i):
            prime[j] = False
        i += 2
    return prime
 
n = int(input('размер матрицы: '))
 
arr = [[randint(-50,50) for i in range(n)] for j in range(n)]
for i in arr :
    print(*map('{:2d}'.format, i))
print()
amax = []
arr = list(zip(*arr))
k = 0
for i in range(n) :
    amax.append(max(arr[i]))
    if arr[i][i] < 0 :
        k = 1
print(*amax)
print()
arr = list(map(list,zip(*arr)))
if k :
    pr = prnum(max(amax))
    for i in range(n) :
        for j in range(n) :
            if pr[abs(arr[i][j])] :
                arr[i][j] = amax[j]
for i in arr :
    print(*map('{:2d}'.format, i))

from random import randint
 
def isPrime(a):
    if a > 1:
        return all(a % i for i in range(2, a))
    else:
        return False
 
r = int(input())
c = int(input())
condition = False
 
matrix = [[randint(-3, 10) for column in range(c)] for row in range(r)] # Заполнение матрицы целыми числами из промежутка [-3, 10]
for row in matrix:
    print(row)
for i in range(c):
    if i < r:
        if matrix[i][i] < 0:
            condition = True
            break
columns = [[matrix[k][i] for k in range(r)] for i in range(c)]
if condition:
    for column in columns:
        for i in range(r):
            if isPrime(column[i]):
                column[i] = max(column)
print()
for i in range(r):
    print([[columns[k][i] for k in range(r)] for i in range(c)][i])