numpy.triu_indices_from¶

numpy.triu_indices_from(arr, k=0)[source]¶

Return the indices for the upper-triangle of a (N, N) array.

See triu_indices for full details.

Parameters :

arr : ndarray, shape(N, N)

The indices will be valid for square arrays.

k : int, optional

Diagonal offset (see triu for details).

Returns :

triu_indices_from : tuple, shape(2) of ndarray, shape(N)

Indices for the upper-triangle of arr.