使用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对oracle进行简单性能测试

    用python对oracle进行简单性能测试

    这篇文章主要介绍了用python对oracle进行简单性能测试的示例,帮助大家更好的理解和使用python,感兴趣的朋友可以了解下
    2020-12-12
  • Python基础之字符串常见操作经典实例详解

    Python基础之字符串常见操作经典实例详解

    这篇文章主要介绍了Python基础之字符串常见操作,结合实例形式详细分析了Python字符串操作基本函数、功能、使用方法及操作注意事项,需要的朋友可以参考下
    2020-02-02
  • python面向对象法实现图书管理系统

    python面向对象法实现图书管理系统

    这篇文章主要为大家详细介绍了python面向对象法实现图书管理系统,具有一定的参考价值,感兴趣的小伙伴们可以参考一下
    2019-04-04
  • Python图像处理二值化方法实例汇总

    Python图像处理二值化方法实例汇总

    这篇文章主要介绍了Python图像处理二值化方法实例汇总,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下
    2020-07-07
  • Python StrEnum基本概念和使用场景分析

    Python StrEnum基本概念和使用场景分析

    StrEnum是Python枚举家族的一个强大补充,特别适合处理字符串常量,它结合了枚举的类型安全性和字符串的灵活性,使得在许多场景下的编程变得更加简洁和安全,本文将介绍StrEnum的基本概念和使用场景,并通过示例代码来展示它的实际应用,感兴趣的朋友跟随小编一起看看吧
    2024-07-07
  • Python爬取附近餐馆信息代码示例

    Python爬取附近餐馆信息代码示例

    这篇文章主要介绍了Python爬取附近餐馆信息代码示例,具有一定借鉴价值,需要的朋友可以参考下。
    2017-12-12
  • Python中三个不可思议的返回功能分享

    Python中三个不可思议的返回功能分享

    这篇文章主要给大家介绍了Python中三个不可思议的返回功能,文中通过实例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下
    2022-01-01
  • Python基础教程之异常详解

    Python基础教程之异常详解

    调试Python程序时,经常会报出一些异常,下面这篇文章就来给大家介绍了关于Python基础教程之异常的相关资料,文中通过示例代码介绍的非常详细,需要的朋友可以参考下
    2019-01-01
  • Python+unittest+requests 接口自动化测试框架搭建教程

    Python+unittest+requests 接口自动化测试框架搭建教程

    这篇文章主要介绍了Python+unittest+requests 接口自动化测试框架搭建教程,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友们下面随着小编来一起学习学习吧
    2020-10-10
  • Python THREADING模块中的JOIN()方法深入理解

    Python THREADING模块中的JOIN()方法深入理解

    这篇文章主要介绍了Python THREADING模块中的JOIN()方法深入理解,本文用简洁易懂的语言总结了对JOIN()方法的理解,不同于其它文章,需要的朋友可以参考下
    2015-02-02

最新评论