numpy.tril_indices_from#
- numpy.tril_indices_from(arr, k=0)[source]#
返回 arr 下三角的索引。
有关详细信息,请参阅
tril_indices
。- 参数:
- arrarray_like
这些索引对于维度与 arr 相同的方形数组有效。
- kint,可选
对角线偏移量(有关详细信息,请参阅
tril
)。
注释
版本 1.4.0 中新增。
示例
>>> import numpy as np
创建一个 4x4 数组
>>> 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]])
将数组传递给函数以获取下三角元素的索引。
>>> trili = np.tril_indices_from(a) >>> trili (array([0, 1, 1, 2, 2, 2, 3, 3, 3, 3]), array([0, 0, 1, 0, 1, 2, 0, 1, 2, 3]))
>>> a[trili] array([ 0, 4, 5, 8, 9, 10, 12, 13, 14, 15])
这是 tril_indices() 的语法糖。
>>> np.tril_indices(a.shape[0]) (array([0, 1, 1, 2, 2, 2, 3, 3, 3, 3]), array([0, 0, 1, 0, 1, 2, 0, 1, 2, 3]))
使用 k 参数返回下三角数组(直至第 k 对角线)的索引。
>>> trili1 = np.tril_indices_from(a, k=1) >>> a[trili1] array([ 0, 1, 4, 5, 6, 8, 9, 10, 11, 12, 13, 14, 15])