Untitled

def get_diagonals(matrix):
    n = matrix.shape[0]
    diagonals = []
    for i in range(n):
        diagonals.append(np.diag(matrix, i))
        diagonals.append(np.diag(matrix[:, ::-1], i))
    return [d.tolist() for d in diagonals if len(d)==n]