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