Создание массива в виде матрицы и дальнейший вывод этого массива в виде массива с преобразованием

n=int(input("кол-во строк: "))
m=int(input("кол-во столбцов: "))
A=[]
for i in range(n):
    row=[]
    for j in range(m):
        row.append(int(input("эл-т матрицы: ")))
    A.append(row)
 
for i in range(n):
    for j in range(m):
        print(f"{A[i][j]:5d}",end="")
    print()

def isPrime(n):
    if n==1:
        return False
    if n==2:
        return True
    if n%2==0:
        return False
    k=3
    while(k*k<=n):
        if n%k==0:
            return False
        k+=2
    return True
    
_=input()
arr=list(map(int,input().split()))
res=list(filter(isPrime,arr))
if len(res)==0:
    print(0)
else:
    print(*res)

def isPrime(n):
    if n == 1:
        return False
    if n == 2:
        return True
    if n % 2 == 0:
        return False
    k = 3
    while k * k <= n:
        if n % k == 0:
            return False
        k += 2
    return True
 
 
n = int(input("кол-во строк: "))
m = int(input("кол-во столбцов: "))
A = [[int(input('>>> ')) for i in range(m)] for j in range(n)]
 
for i in range(n):
    for j in range(m):
        print(f"{A[i][j]:5d}", end="")
    print()
 
res = list(filter(isPrime, sum(A, [])))
if len(res) == 0:
    print(0)
else:
    print(*sorted(res, reverse=True))