Check if Matrix Is X-Matrix LeetCode solution in Python

1
class Solution:
2
    def checkXMatrix(self, grid: List[List[int]]) -> bool:
3
        n = len(grid)
4
        for i in range(n):
5
            for j in range(n):
6
                if i == j or (i + j) == n - 1:
7
                    if grid[i][j] == 0:
8
                        return False
9
                elif grid[i][j] != 0:
10
                    return False
11
        return True