numpy.triu_indices_from#

numpy.triu_indices_from(arr, k=0)[源代码]#

返回 arr 的上三角部分的索引.

请参阅 triu_indices 以获取完整详情.

参数:
arrndarray, 形状(N, N)

索引将对正方形数组有效.

kint, 可选

对角偏移（详见 triu）.

返回:
triu_indices_fromtuple, shape(2) 的 ndarray, shape(N)

arr 上三角的索引.

参见

triu_indices, triu, tril_indices_from

备注

在 1.4.0 版本加入.

示例

>>> import numpy as np

创建一个 4 x 4 的数组

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

传递数组以获取上三角元素的索引.

>>> triui = np.triu_indices_from(a)
>>> triui
(array([0, 0, 0, 0, 1, 1, 1, 2, 2, 3]), array([0, 1, 2, 3, 1, 2, 3, 2, 3, 3]))

>>> a[triui]
array([ 0,  1,  2,  3,  5,  6,  7, 10, 11, 15])

这是 triu_indices() 的语法糖.

>>> np.triu_indices(a.shape[0])
(array([0, 0, 0, 0, 1, 1, 1, 2, 2, 3]), array([0, 1, 2, 3, 1, 2, 3, 2, 3, 3]))

使用 k 参数返回从第 k 个对角线开始的右上三角数组的索引.

>>> triuim1 = np.triu_indices_from(a, k=1)
>>> a[triuim1]
array([ 1,  2,  3,  6,  7, 11])
On this page