numpy.
tril_indices
返回(n,m)数组下三角的索引。
返回的索引将对其有效的数组的行维度。
对角线偏移(参见 tril 详情请参阅。
tril
1.9.0 新版功能.
返回的数组将对其有效的数组的列维度。默认情况下 m 等于 n .
三角形的索引。返回的元组包含两个数组,每个数组的索引沿着数组的一维。
参见
triu_indices
类似的函数,用于上三角形。
mask_indices
接受任意屏蔽函数的通用函数。
triu
笔记
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]])