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。非經特殊聲明，原始代碼版權歸原作者所有，本譯文未經允許或授權，請勿轉載或複製。