【Python/Numpy】numpy的flatnonzero()函数

flatnonzero(x)
首先将x矩阵展平，
然后返回输入的x矩阵的非零元素所在的位置

import numpy as np
x = np.ones((2,3))
print(x)
print(np.flatnonzero(x))
#输出
[[1. 1. 1.]
 [1. 1. 1.]]
[0 1 2 3 4 5]

import numpy as np

x = np.ones((2,3))
print(x)
x[0][0]=0
print(x)
print(np.flatnonzero(x))
#输出
[[1. 1. 1.]
 [1. 1. 1.]]
[[0. 1. 1.]
 [1. 1. 1.]]
[1 2 3 4 5]