xorbits.numpy.tril_indices#

xorbits.numpy.tril_indices(n, k=0, m=None)#

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

参数

n (int) – The row dimension of the arrays for which the returned indices will be valid.
k (int, optional) – Diagonal offset (see tril for details).
m (int, optional) –

1.9.0(numpy) 新版功能.

The column dimension of the arrays for which the returned arrays will be valid. By default m is taken equal to n.

返回

inds – The indices for the triangle. The returned tuple contains two arrays, each with the indices along one dimension of the array.

返回类型

tuple of arrays

参见

triu_indices: similar function, for upper-triangular.
mask_indices: generic function accepting an arbitrary mask function.

tril, triu

提示

1.4.0(numpy) 新版功能.

实际案例

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

>>> il1 = np.tril_indices(4)  
>>> il2 = np.tril_indices(4, 2)  

Here is how they can be used with a sample array:

>>> a = np.arange(16).reshape(4, 4)  
>>> a  
array([[ 0,  1,  2,  3],
       [ 4,  5,  6,  7],
       [ 8,  9, 10, 11],
       [12, 13, 14, 15]])

Both for indexing:

>>> a[il1]  
array([ 0,  4,  5, ..., 13, 14, 15])

And for assigning values:

>>> a[il1] = -1  
>>> a  
array([[-1,  1,  2,  3],
       [-1, -1,  6,  7],
       [-1, -1, -1, 11],
       [-1, -1, -1, -1]])

These cover almost the whole array (two diagonals right of the main one):

>>> a[il2] = -10  
>>> a  
array([[-10, -10, -10,   3],
       [-10, -10, -10, -10],
       [-10, -10, -10, -10],
       [-10, -10, -10, -10]])

警告

This method has not been implemented yet. Xorbits will try to execute it with numpy.

This docstring was copied from numpy.