Untitled
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...