`numpy:triu_indices`

source: /dev/numpy/build-install/usr/lib/python3.14/site-packages/numpy/lib/_twodim_base_impl.py :1054

Signature

def triu_indices ( n , k = 0 , m = None )

Summary

Return the indices for the upper-triangle of an (n, m) array.

Parameters

n : int: The size of the arrays for which the returned indices will be valid.
k : int, optional: Diagonal offset (see triu for details).
m : int, optional: The column dimension of the arrays for which the returned arrays will be valid. By default m is taken equal to n.

Returns

inds : tuple, shape(2) of ndarrays, shape(`n`): The row and column indices, respectively. The row indices are sorted in non-decreasing order, and the corresponding column indices are strictly increasing for each row.

Examples

import numpy as np

✓

Compute two different sets of indices to access 4x4 arrays, one for the upper triangular part starting at the main diagonal, and one starting two diagonals further right:

iu1 = np.triu_indices(4)
iu1

✓

Note that row indices (first array) are non-decreasing, and the corresponding column indices (second array) are strictly increasing for each row. Here is how they can be used with a sample array:

a = np.arange(16).reshape(4, 4)
a

✓

Both for indexing:

a[iu1]

✓

And for assigning values:

a[iu1] = -1
a

✓

These cover only a small part of the whole array (two diagonals right of the main one):

iu2 = np.triu_indices(4, 2)
a[iu2] = -10
a

✓

Aliases