1. 程式人生 > >np.nonzero()函式用法

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