Сортировка вложеного списка по условию

try:
    print('Введите размер матрицы N x M (2 <= n <= 15, 2 <= m <= 13,)')
    n=int(input('Введите количество строк, n='))
    m=int(input('Введите количество столбцов, m='))
    if n >= 2 and n <= 15 and m >= 2 and m <= 13:
        a=[]
        for i in range(n):
            b=[]
            for j in range(m):
                x=int(input('a[{},{}]='.format(i, j)))
                b.append(x)
            a.append(b)
        #вывод массива
        for row in a:
            for elem in row:
                print(elem, end=' ')
            print()
 
        #сортировка
                
        for i in range(n):
            if a[i][0] > a[i][m-1]:
                #print(a[i][0], '>', a[i][m-1], 'Первый элемент больше последнего')
                a[i:].sort()
 
        #вывод после обработки
        for i in range(n):
            for j in range(m):
                print(a[i][j], sep=' ', end=' ')
            print()
    else:
        print('Размер матрицы введен неправильно.')
except Exception:
    print('Ошибка!')

arr = [[3, 2, 1], [6, 5, 4], [9, 8, 7]]
arr = list(map(lambda x: sorted(x) if x[0] > x[-1] else x, arr))
[print(*i) for i in arr]

try:
    print('Введите размер матрицы N x M (2 <= n <= 15, 2 <= m <= 13,)')
    n = int(input('Введите количество строк, n='))
    m = int(input('Введите количество столбцов, m='))
    if n >= 2 and n <= 15 and m >= 2 and m <= 13:
        a = []
        for i in range(n):
            b = []
            for j in range(m):
                x = int(input('a[{},{}]='.format(i, j)))
                b.append(x)
            a.append(b)
        # вывод массива
        for row in a:
            for elem in row:
                print(elem, end=' ')
            print()
 
        # сортировка
        a = list(map(lambda x: sorted(x, reverse=True) if x[0] > x[-1] else x, a))
 
        # вывод после обработки
        [print(*i) for i in a]
    else:
        print('Размер матрицы введен неправильно.')
except Exception:
    print('Ошибка!')

import random
 
 
def get_square_array_fill_rnd_nums(dim: int):
    square_array = []
    for r in range(dim):
        row = []
        square_array.append(row)
        for c in range(dim):
            square_array[r].append(random.randint(1, 9))
    return square_array
 
 
def print_array(arr):
    for r in range(len(arr)):
        print(arr[r])
 
 
sq_arr = get_square_array_fill_rnd_nums(int(input()))
print_array(sq_arr)