Python numpy.argwhere()用法及代码示例

numpy.argwhere()函数用于查找按元素分组的非零数组元素的索引。

用法： numpy.argwhere(arr)

参数：
arr :[数组]输入数组。

Return :[ndarray]非零元素的索引。索引按元素分组。

代码1：

# Python program explaining 
# argwhere() function 
   
import numpy as geek 
  
# input array 
in_arr = [[ 2, 0, 7], [ 0, 5, 9]] 
print ("Input array:", in_arr)  
  
out_arr = geek.argwhere(in_arr) 
print ("Output indices of non zero array element:\n", out_arr)

输出：

Input array: [[2, 0, 7], [0, 5, 9]]
Output indices of non zero array element:
 [[0 0]
 [0 2]
 [1 1]
 [1 2]]

代码2：

# Python program explaining 
# argwhere() function 
   
import numpy as geek 
  
# input array 
in_arr = geek.arange(8).reshape(4, 2) 
print ("Input array:", in_arr)  
  
out_arr = geek.argwhere(in_arr>4) 
print ("Output indices greater than 4:\n", out_arr)

输出：

Input array: [[0 1]
 [2 3]
 [4 5]
 [6 7]]
Output indices greater than 4:
 [[2 1]
 [3 0]
 [3 1]]

相关用法

注：本文由纯净天空筛选整理自jana_sayantan大神的英文原创作品 numpy.argwhere() in Python。非经特殊声明，原始代码版权归原作者所有，本译文未经允许或授权，请勿转载或复制。