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

例子

一维数组

一维数组已经是平坦的，因此这相当于仅应用 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] ，然后提取非零值的索引。