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

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


用法:

dask.array.flatnonzero(a)

在 a 的扁平化版本中返回非零索引。

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

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

這相當於 np.nonzero(np.ravel(a))[0]。

參數：

a：array_like

輸入數據。

返回：

res：ndarray

輸出數組，包含 a.ravel() 的非零元素的索引。

例子：

>>> x = np.arange(-2, 3)  
>>> x  
array([-2, -1,  0,  1,  2])
>>> np.flatnonzero(x)  
array([0, 1, 3, 4])

使用非零元素的索引作為索引數組來提取這些元素：

>>> x.ravel()[np.flatnonzero(x)]  
array([-2, -1,  1,  2])

相關用法


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