numpy.flatnonzero()function用于计算在arr的扁平版本中非零的索引。
用法：numpy.flatnonzero(arr)
参数：
arr :[数组]输入数组。
Return :ndarray
输出数组，其中包含arr.ravel()元素的索引(非零)。
代码1：工作中
# Python program explaining
# flatnonzero() function
import numpy as geek
arr = geek.arange(-3, 4)
print ("Input array:", arr)
out_arr = geek.flatnonzero(arr)
print ("Indices of non zero elements:", out_arr) 输出：
Input array: [-3 -2 -1 0 1 2 3] Indices of non zero elements: [0 1 2 4 5 6]
代码2：使用非零元素的索引作为索引数组。
# Python program using the indices of the non-zero
# elements as an index array to extract these elements
out_arr = arr.ravel()[geek.flatnonzero(arr)]
print ("Output array of non-zero number:", out_arr) 输出：
Output array of non-zero number: [-3 -2 -1 1 2 3]
相关用法
注：本文由纯净天空筛选整理自jana_sayantan大神的英文原创作品 numpy.flatnonzero() in Python。非经特殊声明，原始代码版权归原作者所有，本译文未经允许或授权，请勿转载或复制。