test_array = np.asarray([[1,2,3],[4,5,6],[7,8,9]])

print(test_array>5)

輸出結果是

[[False False False]
 [False False  True]
 [ True  True  True]]

print(np.nonzero(test_array>5))

結果是

(array([1, 2, 2, 2], dtype=int32), array([2, 0, 1, 2], dtype=int32))

也就是生成一個tuple, 有兩個陣列構成。第一個陣列是非零（false）的元素的行索引。第二個陣列是非零（false）的元素的列索引。也就是（1，2），(2，0)，（2，1），（2，2）位置對應的元素非零的

索引從第0列開始，如果只比較第1列上的值：

print(test_array[:,1]>5)

[False False  True]

print(np.nonzero(test_array[:,1]>5))

(array([2], dtype=int32),)

nonzero返回的還是一個tuple元組，但是隻有一個元素

比較常用的是有兩個陣列。 根據一個條件判斷其中一個數組的元素， 根據判斷結果，提取另一個數組的元素。這兩個陣列可以是相同的。

test_array = np.asarray([[1, 2, 3], 

                         [4, 5, 6], 

                         [7
 
, 8, 9]])

index_array = np.asarray([[0, 2, 0], 

                          [0, 0, 0],

                          [0, 2, 2]])

例子A：找到index_array中所有第1列不為0的行，找到第0行和第2行，

從test_array中提取對應的行。

condition_index= np.nonzero( index_array [:,1] !=0 )
print(condition_index)
print( test_array[condition_index] )

輸出：

 [7 8 9]]

另一個例子：找到index_array中所有不為0的元素，找到(0,1),(2,1),(2,2). 從test_array中提取對應行列的元素

condition_index= np.nonzero( index_array != 0 )  
print(condition_index)
print( test_array[condition_index] )

輸出：