dask.array.tril_indices_from
dask.array.tril_indices_from¶
- dask.array.tril_indices_from(arr, k=0)[源代码]¶
返回 arr 的下三角部分的索引。
此文档字符串是从 numpy.tril_indices_from 复制的。
Dask 版本可能存在一些不一致性。
详情请参见 tril_indices。
- 参数
- arrarray_like
索引将对维度与 arr 相同的方形数组有效。
- kint, 可选
对角偏移（详见 tril）。
注释
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]])
传递数组以获取下三角元素的索引。
>>> trili = np.tril_indices_from(a) >>> trili (array([0, 1, 1, 2, 2, 2, 3, 3, 3, 3]), array([0, 0, 1, 0, 1, 2, 0, 1, 2, 3]))
>>> a[trili] array([ 0, 4, 5, 8, 9, 10, 12, 13, 14, 15])
这是 tril_indices() 的语法糖。
>>> np.tril_indices(a.shape[0]) (array([0, 1, 1, 2, 2, 2, 3, 3, 3, 3]), array([0, 0, 1, 0, 1, 2, 0, 1, 2, 3]))
使用 k 参数返回下三角数组到第 k 条对角线的索引。
>>> trili1 = np.tril_indices_from(a, k=1) >>> a[trili1] array([ 0, 1, 4, 5, 6, 8, 9, 10, 11, 12, 13, 14, 15])