np.flatnonzero

python 计算机视觉 numpy

numpy.flatnonzero(a)

• a：输入 array
• 返回：一个数组，该数组是 a 的非 0 索引

该函数输入一个矩阵，返回 array 的非 0 索引（也可以指定索引元素，下面会展示）

Return indices that are non-zero in the flattened version of a.

import numpy as np

x = np.arange(-2 ,3)
print(x)
y = np.flatnonzero(x)
print(y)
y_ = x[np.flatnonzero(x)]
print(y_)
'''
[-2 -1  0  1  2]
[0 1 3 4]
[-2 -1  1  2]
'''


# also can use like this
xx = np.array([1, 3, 2, 3, 4, 3, 1, 2])
yy = np.flatnonzero(xx == 3)
print(yy)
'''
[1 3 5]
'''

本文由 Yonghui Wang 创作，采用 知识共享署名4.0 国际许可协议进行许可
本站文章除注明转载/出处外，均为本站原创或翻译，转载前请务必署名
最后编辑时间为: Dec 19, 2024 12:13 pm