numpy-nonzero

最新推荐文章于 2024-11-09 18:30:00 发布

原创最新推荐文章于 2024-11-09 18:30:00 发布 · 433 阅读

0 ·

CC 4.0 BY-SA版权

python-numpy 专栏收录该内容

3 篇文章

订阅专栏

本文详细介绍了numpy中nonzero函数的应用，展示了如何通过该函数找出数组中的非零元素及其位置。对于array和mat类型的数据结构，nonzero函数返回值有所不同：前者返回单一数组，后者返回两个矩阵，分别指示非零元素所在的行和列。

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值；行放在第一个里面，例放在第二个里面