【python学习笔记】numpy.nonzero()函数

Numpy是Python的一个科学计算的库，提供了矩阵运算的功能,下面讲解一下numpy.nonzero()函数用法。

numpy函数

输入值：数组或矩阵

返回输入值中非零元素的信息（以矩阵的形式）

这些信息中包括 两个矩阵， 包含了相应维度上非零元素所在的行标号，与列标标号。

例如：a=mat([ [1,0,0],[0,0,0],[0,0,0]])

则 nonzero(a) 

返回值为两个矩阵：(matrix([[0]], dtype=int32), matrix([[0]], dtype=int32)) , 

表示输入矩阵a只有1个非零值， 第一个matrix([[0]], dtype=int32)表示非零元素在第0行， 第二个matrix([[0]], dtype=int32)表示在第0行的第0列。

再比如 a=mat([[1,0,0],[1,0,0],[0,0,0]])

则 nonzero(a) 

返回值为(matrix([[0, 1, 2]], dtype=int32), matrix([[0, 0, 2]], dtype=int32)),

因为矩阵a只有3个非零值， 所以返回值行维度数据matrix([[0, 1, 2]], dtype=int32)表示非零元素出现在a中的第0行、第1行和第2行，返回值列维度数据matrix([[0, 0, 2]], dtype=int32)对应表示每行中第几列，即第0行第0列，第1行第0例，第2行第2列。

from numpy import *

a = mat([[1,0,0],[0,0,0],[0,0,0]])

print(a)

print(nonzero(a))

a = mat([[1,0,0],[1,0,0],[0,0,1]])

print(a)

print(nonzero(a))

a = mat([[1,0,0],[1,0,0],[0,0,1]])

print(a)

print(nonzero(a))

a = mat([[1,0,0],[1,0,0],[1,1,1]])

print(a)

print(nonzero(a))

程序运行结果为：

[[1 0 0]

 [0 0 0]

 [0 0 0]]

(matrix([[0]], dtype=int32), matrix([[0]], dtype=int32))

[[1 0 0]

 [1 0 0]

 [0 0 1]]

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

[[1 0 0]

 [1 0 0]

 [0 0 1]]

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

[[1 0 0]

 [1 0 0]

 [1 1 1]]

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