Untitled

 avatar
unknown
plain_text
2 years ago
253 B
5
Indexable
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]
Editor is loading...