📌  相关文章
📜  Python中的numpy.argmin

📅  最后修改于: 2020-06-07 05:20:31             🧑  作者: Mango

numpy.argmin(array, axis = None, out = None) : 返回特定轴上数组min元素的索引。

参数：

array：要处理的输入数组
axis：[int，可选]沿指定的轴，如0或1
out：[array optional]提供将输出插入到out的功能
数组，并且应该具有适当的形状和dtype

返回：

指向数组的索引数组，其形状与array.shape相同，但沿轴的尺寸已删除。

代码1：

# Python程序说明argmin()的工作 
  
import numpy as geek  
  
# 处理一维阵列
array = geek.arange(8) 
print("输入数组 : \n", array) 
  
  
# 根据索引返回min元素的索引 
print("\n最小元素的指标 : ", geek.argmin(array, axis=0))

输出：

输入数组：
 [0 1 2 3 4 5 6 7] 

最小元素的指标：0

代码2：

# Python程序说明argmin()的工作
  
import numpy as geek  
  
# 处理二维阵列 
array =  geek.random.randint(16, size=(4, 4)) 
print("INPUT ARRAY : \n", array) 
  
# 根据索引返回min元素的索引 
  
'''    
   [[ 8 13  5  0] 
   [ 0  2  5  3] 
   [10  7 15 15] 
   [ 3 11  4 12]] 
     ^  ^  ^  ^ 
     0  2  4  0  - element 
     1  1  3  0  - indices 
'''
print("\n最小元素的索引 : ", geek.argmin(array, axis = 0))

输出：

输入数组：
 [[8 13 5 0] 
 [0 2 5 3] 
 [10 7 15 15] 
 [3 11 4 12]] 

最小元素的索引：[1 1 3 0]

代码3：

# Python程序说明argmin()的工作
  
import numpy as geek  
  
# 处理二维阵列 
array =  geek.arange(10).reshape(2, 5) 
print("数组 : \n", array) 
  
array[0][0] = 10
array[1][1] = 1
array[0][1] = 1
print("\n数组 : \n", array) 
       
# 返回最小元素 
print("\n数组 : ", geek.argmin(array)) 
  
# 最小元素的首次出现 
print("\n最早元素索引 : ", geek.argmin(array, axis = 0))

输出：

数组：
 [[0 1 2 3 4] 
 [5 6 7 8 9]] 

数组：
 [[10 1 2 3 4] 
 [5 1 7 8 9]] 

数组：1 

最早元素索引：[1 0 0 0 0]