numpy.nonzero()函数

调用方式：

numpy.nonzero(a)

各个参数意义

1. a：输入数组
2. 有返回值，返回值为一个元组类型的索引数组，即得到 非零元素在原数组中的下标

代码如下：

# -*- coding: utf-8 -*-
"""
Created on Sat Oct 17 06:27:33 2020

@author: 15025
"""

import numpy as np


class NumpyStudy:
    def printNumpyVersion(self):
        array1 = np.array([0, 1, 2, 6])
        array2 = np.array([True, False, False, True])
        array_output1 = np.nonzero(array1)
        array_output2 = np.nonzero(array2)
        print("数组array_output1的值为: ")
        print(array_output1)
        print("数组array_output2的值为: ")
        print(array_output2)
        

if __name__ == "__main__":
    main = NumpyStudy()
    main.printNumpyVersion()
"""
数组array_output1的值为: 
(array([1, 2, 3], dtype=int64),)
数组array_output2的值为: 
(array([0, 3], dtype=int64),)
"""

结果显而易见，这里不做过多说明。