Python中的nonzero()函数

Python中的nonzero()函数

numpy.nonzero（a）函数一般返回两行array（），第一个数组是非零元素的行索引，第二个数组是列索引，也就是一个2*N 的矩阵，其中N 表示的是矩阵a中不为0的元素个数。

import numpy as np

C = np.array([[0,5,9],
              [4,0,10]])

print(np.nonzero(C))
'''
(array([0, 0, 1, 1], dtype=int64), array([1, 2, 0, 2], dtype=int64))
'''

矩阵 C中，C[0,1],C[0,2],C[1,0],C[1,2]元素的值非零.