调用方式:
numpy.nonzero(a)
各个参数意义
a
:输入数组- 有返回值,返回值为一个元组类型的索引数组,即得到 非零元素在原数组中的下标
代码如下:
# -*- 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),)
"""
结果显而易见,这里不做过多说明。