dask.array.triu_indices_from
dask.array.triu_indices_from¶
- dask.array.triu_indices_from(arr, k=0)[源代码]¶
返回 arr 的上三角部分的索引。
此文档字符串是从 numpy.triu_indices_from 复制的。
Dask 版本可能存在一些不一致性。
详情请参见 triu_indices。
- 参数
- arrndarray, 形状(N, N)
索引将对方形数组有效。
- kint, 可选
对角偏移（详见 triu）。
- 返回
- triu_indices_fromtuple, shape(2) of ndarray, shape(N)
arr 上三角的索引。
注释
1.4.0 新版功能.
示例
>>> import numpy as np
创建一个 4x4 的数组
>>> 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])