np.nonzero（）函数用法

返回数组中不为0的元素的下标。

数组中元素可为布尔、整型和浮点型，返回值为元祖

一.一维数组

1.数组元素为布尔类型

a=np.array([True,False,True,False])
b=np.nonzero(a)
print(b)

输出结果为(array([0, 2], dtype=int64),)

即下标为0和2的元素值非0

2.数组元素为int类型

a=np.array([2,10,0,0])

3.数组元素为float类型

a=np.array([2.1,10,0.0,0])

二.二维数组

二维数组返回的为长度为2的元祖

a=np.array([[True,False,True],[True,False,False]])
b=np.nonzero(a)
print(b)

返回结果为(array([0, 0, 1], dtype=int64), array([0, 2, 0], dtype=int64))

即下标为（0,0）,(0,2),(1,0)的元素值不为0

 

转载于:https://www.cnblogs.com/wmy-ncut/p/9915503.html