Untitled

def get_diagonals(matrix):
    n = len(matrix)
    diagonals = [[] for _ in range(2*n-1)]
    for i in range(n):
        for j in range(n):
            diagonal_index = i-j+n-1
            diagonals[diagonal_index].append(matrix[i][j])
    return [d for d in diagonals if len(d)==n]