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

Python dask.array.topk用法及代碼示例

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

從給定軸上的 a 中提取 k 個最大的元素，並將它們從大到小排序返回。如果 k 為負數，則改為提取 -k 個最小元素，並將它們從最小到最大排序返回。

當k 遠小於塊大小時，這表現最佳。所有結果將沿給定軸以單個塊的形式返回。

參數：

x: Array：: 正在排序的數據
k: int：
axis: int, optional：
split_every: int >=2, optional：: 請參閱reduce()。當 k 處於與塊大小相同或更大的數量級時，此參數變得非常重要，因為它可以防止將輸入數組的全部或重要部分一次全部放入內存中，並且在以下情況下也會對網絡傳輸產生負麵影響在分布式上運行。

返回：

沿給定軸選擇大小為 abs(k) 的 x。

例子：

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

相關用法

注：本文由純淨天空篩選整理自dask.org大神的英文原創作品 dask.array.topk。非經特殊聲明，原始代碼版權歸原作者所有，本譯文未經允許或授權，請勿轉載或複製。