NumPy参考 >例行程序 >Indexing routines > numpy.tril_indices
numpy.
tril_indices
(n,k = 0,m = None )[源代码] ¶返回(n,m)数组下三角的索引。
返回索引对其有效的数组的行维。
对角线偏移(tril
有关详细信息,请参见)。
1.9.0版中的新功能。
返回数组将对其有效的数组的列维。默认情况下,m等于n。
三角形的索引。返回的元组包含两个数组,每个数组的索引沿数组的一维。
笔记
1.4.0版的新功能。
例子
计算两组不同的索引以访问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]])