numpy中nonzero函数详解

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]))