numpy.tril_indices_from

numpy.tril_indices_from(arr, k=0)

傳回 arr 下三角的索引。

參見 tril_indices 以取得完整細節。

參數:

arrarray_like

索引對於維度與 arr 相同的方形陣列有效。

kint，選用

對角線偏移量 (參見 tril 以取得細節)。

另請參閱

tril_indices、tril、triu_indices_from

範例

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