Работа 12 Python

import random

n = int(input("Введите количество строк: "))
m = int(input("Введите количество столбцов: "))
a = []
b = []
b2 = []
c = []
buff = 0
buff2 = 0
if m != n:
    print("Недопустимые значения!")

else:
    if n >= 2 and n <=5 and m >= 2 and m <= 5:
        for i in range(n):                      # создаём матрицу
            a.append([0] * m)

        r = str(input("Заполнить матрицу случайными значениями? "))

        if r == "нет":
            for i in range(n):
                for j in range(m):
                    a[i][j] = int(input("Введите значение матрицы: "))  # заполняем матрицу

            for i in range(len(a)):         # len(A) - возвращает количество строк в матрице А
                for j in range(len(a[i])):  # len(A[i]) - возвращает количество элементов в строке i
                    print(a[i][j], end = ' ')
                print()                     # делаем переход на новую строку

            for i in range(0, n):
                buff = 0
                for j in range (0, m):
                    buff += a[i][j]     # считаем сумму строк матрицы
                b.append(buff)

            for i in range(0, n):
                buff2 = 0
                for j in range (0, m):
                    buff2 += a[j][i] # считаем сумму столбцов матрицы
                b2.append(buff2)

            b2.sort()  # Сортируем по возрастанию
            b.sort(reverse=True)   # Сортируем по убыванию

            c = b + b2 # Массив с суммой строк и столбцов
            print(c) # Ответ

        elif r == "да":
            for i in range(0, n):
                for j in range(0, m):
                    a[i][j] = random.randint(1, 100) # заполняем матрицу рандомными числами от 1 до 100

            for i in range(len(a)):         # len(A) - возвращает количество строк в матрице А
                for j in range(len(a[i])):  # len(A[i]) - возвращает количество элементов в строке i
                    print(a[i][j], end = ' ')
                print()                     # делаем переход на новую строку

            for i in range(0, n):
                buff = 0
                for j in range (0, m):
                    buff += a[i][j] # считаем сумму строк матрицы
                b.append(buff)

            for i in range(0, n):
                buff2 = 0
                for j in range (0, m):
                    buff2 += a[j][i] # считаем сумму столбцов матрицы
                b2.append(buff2)

            b2.sort()  # Сортируем по возрастанию
            b.sort(reverse=True)   # Сортируем по убыванию

            c = b + b2 # Массив с суммой строк и столбцов
            print(c) # Ответ

        else:
            print("Ответ должен быть только 'да' или 'нет'!")

    else:
        print("Недопустимые значения")