numpy中nonzero函数详解

最新推荐文章于 2024-04-23 11:34:10 发布

醉小义

最新推荐文章于 2024-04-23 11:34:10 发布

阅读量724

点赞数

CC 4.0 BY-SA版权

分类专栏： python 文章标签： nonzero python

本文链接：https://blog.youkuaiyun.com/qq_30638831/article/details/80481709

python 专栏收录该内容

63 篇文章

订阅专栏

本文介绍了如何使用Python中的NumPy库来确定数组中非零元素的位置。通过具体实例展示了np.nonzero()函数的用法，并给出了如何找到特定值所在位置的方法。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

import numpy as np

'''
    nonzero()
        获取元素的行下标与列下标
'''

a = np.array([[1, 0, 3],
              [4, 5, 0]])

print(np.nonzero(a))
'''
    输出结果:
        (array([0, 0, 1, 1]), array([0, 2, 0, 1]))

        array([0, 0, 1, 1]) 代表a所在的行标号 ,从第一行，第二行，依次进行非零的元素下标
        array([0, 2, 0, 1]  代表a所在的列标号 ,从第一列，第二列，依次进行非零的元素下标


'''

# 应用   输出a中所有元素等于5所在的行与列

print(np.nonzero(a == 5))  # (array([1]), array([1]))