Помогите пожалуйста решить задачу на Python

[Dizaster]

Дана квадратная матрица, необходимо проверить, является ли она единичной.

мой код, который не работает:

matrix = [[1, 0, 0, 0],
          [0, 1, 0, 0],
          [0, 0, 1, 0],
          [0, 0, 0, 1]]

for i in (matrix):
    for j in (i, matrix):
        if (i == j and matrix[i][j] != 1) or (i != j and matrix[i][j] != 0):
             print('Матрица не единичная')
        else:
             print('Матрица единичная')

[p51x]

def main():
    matrix = [[1, 0, 0, 0],
    [0, 1, 0, 0],
    [0, 0, 1, 0],
    [0, 0, 0, 1]]

    for i in range(len(matrix)):
        for j in range(len(matrix[i])):
            if (i == j and matrix[i][j] != 1) or (i != j and matrix[i][j] != 0):
                print('Матрица не единичная')
                return

    print('Матрица единичная')

if __name__ == '__main__': 
    main()

[BDA]

Еще 2 способа (задействуя else ветку цикла, используя функцию all):

def main():
    matrix = [[1, 0, 0, 0],
    [0, 1, 0, 0],
    [0, 0, 1, 0],
    [0, 0, 0, 1]]

    # 1
    for i in range(len(matrix)):
        for j in range(len(matrix[i])):
            if (matrix[i][j] != int(i == j)):
                print('Матрица не единичная')
                break
        else:
            continue
        break
    else:
        print('Матрица единичная')

    # 2
    if all(matrix[i][j] == int(i == j) for i in range(len(matrix)) for j in range(len(matrix[i]))):
        print('Матрица единичная')
    else:
        print('Матрица не единичная')

if __name__ == '__main__': 
    main()

[ViktorR]

"Красота - это страшная сила!" (Фаина Раневская)