Дан квадратный двумерный массив целых чисел. Найти количество элементов, расположенных в строках с положительным элементом на главной диагонали. написать программу в питоне(через списки). дать пояснение к программе

Python

# - обозначение комментария

# подключаем модуль, чтобы в ручную не вводить матрицу  

from random import randint

# вводим размерность матрицы

n=int(input('Введите n: '))

# заполняем матрицу случайными числами в диапазоне (-20,20)

a=[[randint(-20,20) for j in range(n)] for i in range(n)]  

# печатаем матрицу построчно, чтобы было понятно и красиво

for i in a: print(i)

# обнуляем счётчик, который будет считать количество элементов

s=0

# запускаем цикл  

for i in range (n):

# проверяем элемент главной диагонали матрицы.

      if a[i][i]>0:

# если он положительный увеличиваем счётчик на количество элементов в строке матрицы

      s+=n

# выводим количество подсчитанных элементов

print(s)