numpy.flatnonzero #
- 麻木的。flatnonzero ( a ) [来源] #
返回 a 的展平版本中非零的索引。
这相当于
np.nonzero(np.ravel(a))[0]
.- 参数:
- 类似数组
输入数据。
- 返回:
- 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])