当前位置: 首页>>代码示例 >>用法及示例精选 >>正文


Python NumPy flatnonzero方法用法及代码示例


Numpy 的 flatnonzero(~) 方法返回输入数组的扁平版本中非零值的索引。

参数

1. a | array_like

输入数组。

返回值

输入数组的扁平版本中非零值索引的 Numpy 数组。

例子

一维数组

一维数组已经是平坦的,因此这相当于仅应用 np.nonzero(~)

np.flatnonzero([5,3,0,2])



array([0, 1, 3])

这里,由于值 5、3 和 2 是非零,因此返回索引 0、1 和 3。

二维数组

假设我们有一个二维数组:

np.flatnonzero([[1,0],[2,3]])



array([0, 2, 3])

在这里,二维数组首先被展平为 [1,0,2,3] ,然后提取非零值的索引。

相关用法


注:本文由纯净天空筛选整理自Isshin Inada大神的英文原创作品 NumPy | flatnonzero method。非经特殊声明,原始代码版权归原作者所有,本译文未经允许或授权,请勿转载或复制。