numpy.triu_indices_from #

麻木的。triu_indices_from ( arr , k = 0 ) [来源] #

返回 arr 上三角形的索引。

请参阅triu_indices了解完整详细信息。

参数
arr ndarray,形状(N,N)

索引对于方形数组有效。

k整数,可选

对角线偏移(triu详情请参见)。

返回
triu_indices_from元组,ndarray 的形状(2),形状(N)

arr上三角的索引。

笔记

1.4.0 版本中的新增功能。

例子

创建一个 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])