def is_identity_element mtr for in xrange for in xrange if mtr or mtr

def is_identity_element (mtr, n):
    for e in xrange (n):
        for j in xrange (n):
            if mtr[e][j] != j or mtr[j][e] != j:
                break
        else:
            return True
    return False