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


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


用法:

dask.array.argwhere(a)

查找按元素分組的非零數組元素的索引。

此文檔字符串是從 numpy.argwhere 複製的。

可能存在與 Dask 版本的一些不一致之處。

參數

aarray_like

輸入數據。

返回

index_array(N, a.ndim) ndarray

非零元素的索引。索引按元素分組。該數組將具有 (N, a.ndim) 形狀,其中 N 是非零項目的數量。

注意

np.argwhere(a)np.transpose(np.nonzero(a)) 幾乎相同,但會為 0D 數組生成正確形狀的結果。

argwhere 的輸出不適合索引數組。為此,請改用nonzero(a)

例子

>>> x = np.arange(6).reshape(2,3)  
>>> x  
array([[0, 1, 2],
       [3, 4, 5]])
>>> np.argwhere(x>1)  
array([[0, 2],
       [1, 0],
       [1, 1],
       [1, 2]])

相關用法


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