numpy.nonzero函数

来源:互联网 发布:2017最新网络综艺节目 编辑:程序博客网 时间:2024/05/21 17:16

nonzero函数返回一个元组,元组的第一个元素为一个数组,存储所有非零元素的行下标,第二个元素也为一个元组,存储相应的列坐标

举个例子:a=mat([1,2,3]

   [0,1,0],

   [1,0,0]

)

nonzero(a)的返回值为([0,0,0,1,2],[0,1,2,1,0]),这些就是1,2,3,1,1这五个非0元素的行下标和列下标

0 0
原创粉丝点击