Python numpy tril_indices用法及代码示例

本文简要介绍 python 语言中 numpy.tril_indices 的用法。

用法:

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

返回 (n, m) 数组的 lower-triangle 的索引。

参数：

n： int

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

k： 整数，可选

对角线偏移(详见 tril )。

m： 整数，可选

返回数组对其有效的数组的列维度。默认情况下，m 等于 n。

返回：

inds： 数组元组

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

注意：

例子：

计算两组不同的索引以访问 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]])