numpy.tril_indices#

numpy.tril_indices(n, k=0, m=None)[source]#

返回 (n, m) 数组下三角形的索引。

参数::
nint

返回索引有效的数组的行维数。

kint, 可选

对角线偏移量(详情请参见 tril)。

mint, 可选

版本 1.9.0 中新增。

返回数组有效的数组的列维数。默认情况下,m 等于 n

返回值::
inds数组元组

三角形的索引。返回的元组包含两个数组,每个数组都包含数组一个维度的索引。

另请参见

triu_indices

类似函数,用于上三角。

mask_indices

接受任意掩码函数的通用函数。

tril, triu

注意

版本 1.4.0 中新增。

示例

>>> import numpy as np

计算两组不同的索引以访问 4x4 数组,一组用于从主对角线开始的下三角部分,另一组用于从主对角线右移两个对角线开始。

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

以下是它们如何与示例数组一起使用

>>> 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]])

用于索引

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

以及用于赋值

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

它们覆盖了几乎整个数组(从主对角线向右移两个对角线)

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