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