import numpy as np

'''
    nonzero()
獲取元素的行下標與列下標'''
a = np.array([[1, 0, 3],
[4, 5, 0]])

print(np.nonzero(a))
'''
輸出結果:
        (array([0, 0, 1, 1]), array([0, 2, 0, 1]))
        array([0, 0, 1, 1]) 代表a所在的行標號 ,從第一行，第二行，依次進行非零的元素下標        array([0, 2, 0, 1]  代表a所在的列標號 ,從第一列，第二列，依次進行非零的元素下標'''
# 應用   輸出a中所有元素等於5所在的行與列
 
print(np.nonzero(a == 5))  # (array([1]), array([1]))