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