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] ，然後提取非零值的索引。