numpy中nonzero函数详解

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

原创最新推荐文章于 2024-04-23 11:34:10 发布 · 756 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#nonzero #python

python 专栏收录该内容

63 篇文章

订阅专栏

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

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]))