Python NumPy中nonzero()函数用法详解
发布时间：2026-10-13 |
6人浏览过
本文介绍Python Num库中nonzero()函数的核心用法。该函数用于返回数组中非零元素的索引坐标，结果以元组形式呈现，分别对应各维度的位置信息。通过示例代码展示了如何提取非零元素的行列坐标及其对应的值，帮助开发者高效处理矩阵数据。
nonzero函数用于返回数组中非零元素的索引位置。
该函数的返回值是一个元组，包含两个数组，分别对应不同维度上非零元素的索引值。
import numpy as np
A = np.mat([[0,1,2,3,4,3,2,1,0],[0,1,2,3,4,5,6,7,0]])
x = A.nonzero() #取出矩阵中的非零元素的坐标
print x #输出是一个元组，两个维度。一一对应，
#返回非零元素在矩阵中的位置,前一个列表存放非零行坐标，后一个列表存放非零元素列坐标
#(array([0, 0, 0, 0, 0, 0, 0, 1, 1, 1, 1, 1, 1, 1]), array([1, 2, 3, 4, 5, 6, 7, 1, 2, 3, 4, 5, 6, 7]))
#放下来一一对应，即
#(array([0, 0, 0, 0, 0, 0, 0, 1, 1, 1, 1, 1, 1, 1]),
#array([1, 2, 3, 4, 5, 6, 7, 1, 2, 3, 4, 5, 6, 7]))
# (0,1) (0,2).............................. (1,6) (1,7) 坐标上的元素都是非0的
print A[x],'\n' #取出矩阵中的非零元素
#[[1 2 3 4 3 2 1 1 2 3 4 5 6 7]]
B = np.mat([[1,1,0,1,0,1,0,0,1],[0,1,1,0,0,0,1,1,1]])
print np.nonzero(B) # 与B.nonzero()等价
以上关于Python NumPy中nonzero()函数的用法介绍到此结束，希望能为读者提供有益参考，并感谢大家的支持。
本站声明
本文内容仅供参考，不代表本站观点。转载或引用请注明出处。如发现内容问题，请通过联系我们页面反馈。
