Python中的 numpy.nonzero()

numpy.nonzero()函数用于计算非零元素的索引。

它返回一个数组元组，arr 的每个维度一个，包含该维度中非零元素的索引。
可以使用arr[nonzero(arr)]获得数组中相应的非零值。要按元素而不是维度对索引进行分组，我们可以使用transpose(nonzero(arr)) 。

Syntax : numpy.nonzero(arr)

Parameters :
arr : [array_like] Input array.

Return : [tuple_of_arrays] Indices of elements that are non-zero.

代码#1：工作

# Python program explaining
# nonzero() function
  
import numpy as geek
arr = geek.array([[0, 8, 0], [7, 0, 0], [-5, 0, 1]])
  
print ("Input  array : \n", arr)
    
out_tpl = geek.nonzero(arr)
print ("Indices of non zero elements : ", out_tpl) 

输出 ：

Input array :
[[ 0 8 0]
[ 7 0 0]
[-5 0 1]]
Indices of non zero elements : (array([0, 1, 2, 2], dtype=int64), array([1, 0, 0, 2], dtype=int64))

代码#2：

# Python program for getting
# The corresponding non-zero values:
out_arr = arr[geek.nonzero(arr)]
  
print ("Output array of non-zero number: ", out_arr) 

输出 ：

Output array of non-zero number:  [ 8  7 -5  1]

代码#3：

# Python program for grouping the indices
# by element, rather than dimension
  
out_ind = geek.transpose(geek.nonzero(arr))
  
print ("indices of non-zero number: \n", out_ind) 

输出 ：

indices of non-zero number: 
 [[0 1]
 [1 0]
 [2 0]
 [2 2]]