python nonzero

nonzero(a)返回數組a中值不爲零的元素的下標，它的返回值是一個長度爲a.ndim(數組a的軸數)的元組，元組的每個元素都是一個整數數組，其值爲非零元素的下標在對應軸上的值。例如對於一維布爾數組b1，nonzero(b1)所得到的是一個長度爲1的元組，它表示b1[0]和b1[2]的值不爲0(False)。

>>> b1=np.array([True, False, True, False])
>>> np.nonzero(b1)
(array([0, 2], dtype=int64),)

對於二維數組b2，nonzero(b2)所得到的是一個長度爲2的元組。它的第0個元素是數組a中值不爲0的元素的第0軸的下標，第1個元素則是第1軸的下標，因此從下面的結果可知b2[0,0]、b[0,2]和b2[1,0]的值不爲0：

>>> b2 = np.array([[True, False, True], [True, False, False]])
>>> np.nonzero(b2)
(array([0, 0, 1], dtype=int64), array([0, 2, 0], dtype=int64))