當前位置: 首頁>>代碼示例 >>用法及示例精選 >>正文

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。非經特殊聲明，原始代碼版權歸原作者所有，本譯文未經允許或授權，請勿轉載或複製。