当前位置: 首页>>代码示例 >>用法及示例精选 >>正文

Python dask.array.argtopk用法及代码示例

用法: dask.array.argtopk(a, k, axis=- 1, split_every=None)

从给定轴上的 a 中提取 k 个最大元素的索引，并将它们从最大到最小排序。如果 k 为负数，则提取 -k 个最小元素的索引，并按从小到大的顺序返回它们。

当k 远小于块大小时，这表现最佳。所有结果将沿给定轴以单个块的形式返回。

参数：

x: Array：: 正在排序的数据
k: int：
axis: int, optional：
split_every: int >=2, optional：: 见topk()。 topk 的性能注意事项也适用于此处。

返回：

沿给定轴选择大小为 abs(k) 的 x 的 np.intp 索引。

例子：

>>> import dask.array as da
>>> x = np.array([5, 1, 3, 6])
>>> d = da.from_array(x, chunks=2)
>>> d.argtopk(2).compute()
array([3, 0])
>>> d.argtopk(-2).compute()
array([1, 2])

相关用法

注：本文由纯净天空筛选整理自dask.org大神的英文原创作品 dask.array.argtopk。非经特殊声明，原始代码版权归原作者所有，本译文未经允许或授权，请勿转载或复制。