使用numpy nonzero 找出非0元素

更新时间：2021年05月14日 10:05:46 作者：ItisNagatoYuki

这篇文章主要介绍了使用numpy nonzero 找出非0元素的方法，具有很好的参考价值，希望对大家有所帮助。如有错误或未考虑完全的地方，望不吝赐教

直接上代码吧~

import numpy as np
a = np.array([[30,40,70],[80,20,10],[50,90,60]])
print(a)
print(np.nonzero(a))

[[30 40 70]
 [80 20 10]
 [50 90 60]]
(array([0, 0, 0, 1, 1, 1, 2, 2, 2], dtype=int64), array([0, 1, 2, 0, 1, 2, 0, 1, 2], dtype=int64))

第一个array为x轴第二个array为y轴

补充：【Numpy学习】python查找矩阵中不为0元素的索引（np.nonzero()）

在用矩阵分解方法做模型时，需要对模型的结果做验证。

在验证过程中需要mask训练集、验证集和测试集。

这时候就需要原矩阵S SS中不为0元素和为0元素的索引值，这个方法在matlab中是find方法，在用python实现时就需要np.nonzero()。

下面看一段代码：

import numpy as np
a = np.arange(12).reshape(3, 4)
print(a)
[[ 0  1  2  3]
 [ 4  5  6  7]
 [ 8  9 10 11]]
print(np.nonzero(a))
(array([0, 0, 0, 1, 1, 1, 1, 2, 2, 2, 2]), array([1, 2, 3, 0, 1, 2, 3, 0, 1, 2, 3]))

值得注意的是np.nonzero(a)输出的是两个array第一个array中的值指的是行，第二个指的是列。

如0,1表明第0行第一列的值不为0。

以上为个人经验，希望能给大家一个参考，也希望大家多多支持脚本之家。

您可能感兴趣的文章:

Python算法之栈（stack）的实现
这篇文章主要介绍了Python算法之栈（stack）的实现,非常实用,需要的朋友可以参考下
2014-08-08
Python可视化神器pyecharts绘制地理图表
这篇文章主要介绍了Python可视化神器pyecharts绘制地理图表，文章围绕主题展开详细的内容介绍，具有一定的参考价值，需要的小伙伴可以参考一下
2022-07-07
Python制作简单的网页爬虫
自己写的一个爬虫，模仿了python核心编程书里的程序，有详细的注释。是我一个理解学习的过程吧。有需要的小伙伴可以参考下
2015-11-11
tensorflow输出权重值和偏差的方法
本篇文章主要介绍了tensorflow输出权重值和偏差的方法，小编觉得挺不错的，现在分享给大家，也给大家做个参考。一起跟随小编过来看看吧
2018-02-02
如何远程使用服务器上的Jupyter notebook
这篇文章主要介绍了如何远程使用服务器上的Jupyter notebook,主要是在服务器端执行操作，需要特别注意为了防止远程中断，使用挂起操作，即执行nohup jupyter notebook，需要的朋友可以参考下
2023-02-02
解决Python中定时任务线程无法自动退出的问题
今天小编就为大家分享一篇解决Python中定时任务线程无法自动退出的问题，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
2019-02-02
python实现机器人行走效果
这篇文章主要为大家详细介绍了python实现机器人行走效果，具有一定的参考价值，感兴趣的小伙伴们可以参考一下
2018-01-01
Python3操作SQL Server数据库(实例讲解)
下面小编就为大家带来一篇Python3操作SQL Server数据库(实例讲解)。小编觉得挺不错的，现在就分享给大家，也给大家做个参考。一起跟随小编过来看看吧
2017-10-10
浅析python中的del用法
python中的del用法比较特殊，新手学习往往产生误解，弄清del的用法，可以帮助深入理解python的内存方面的问题。这篇文章主要介绍了python中的del用法,需要的朋友可以参考下
2020-09-09
YOLOv5目标检测之anchor设定
在训练yolo网络检测目标时,需要根据待检测目标的位置大小分布情况对anchor进行调整,使其检测效果尽可能提高,下面这篇文章主要给大家介绍了关于YOLOv5目标检测之anchor设定的相关资料,需要的朋友可以参考下
2022-05-05