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