dask.array.flatnonzero
dask.array.flatnonzero¶
- dask.array.flatnonzero(a)[源代码]¶
返回在 a 的展平版本中非零的索引。
此文档字符串是从 numpy.flatnonzero 复制而来的。
Dask 版本可能存在一些不一致性。
这等价于
np.nonzero(np.ravel(a))[0]。- 参数
- aarray_like
输入数据。
- 返回
- resndarray
输出数组，包含
a.ravel()中非零元素的索引。
示例
>>> import numpy as np >>> 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])