nonzero返回值为元组, 两个值分别为两个维度, 包含了相应维度上非零元素的目录值。注意array和mat的区别
>>> A
array([[ 0.],
[ 1.],
[ 0.],
[ 0.],
[ 0.]])
>>> nonzero(A) #第1行第0个为非0值
(array([1]), array([0]))
>>> nonzero(A)[0]
#======================================
array([1])
>>> a=mat([[1,0,0],[1,0,0],[0,1,0]])
>>> nonzero(a)
(matrix([[0, 1, 2]]), matrix([[0, 0, 1]]))#第某行某例为非0值;行放在第一个里面,例放在第二个里面
numpy-nonzero
最新推荐文章于 2024-04-23 11:34:10 发布