NumPy 函数

NumPy argmin() 函数返回沿轴的最小值的索引。默认情况下,它是在展平数组上计算的,否则是在指定轴上计算的。

语法

numpy.argmin(a, axis=None, out=None) 

参数

a必填。 指定输入数组。
axis可选。 指定计算最小值索引的一个或多个轴。默认情况下是在展平数组上计算它。
out可选。 指定结果的输出数组。默认值为"无"。如果提供,它必须具有与输出相同的形状。

返回值

当out=None,返回一个包含最小值索引的数组,否则返回对输出数组的引用。

示例:

在下面的示例中,argmin() 函数用于查找整个数组中最小值的索引。

import numpy as np
Arr = np.arange(12).reshape(3,4)

print("Array 为:")
print(Arr)

#最小值索引
idx = np.argmin(Arr)
print("\n最小值指标为:", idx) 

上述代码的输出将是:

Array 为:
[[ 0  1  2  3]
 [ 4  5  6  7]
 [ 8  9 10 11]]

最小值指标为: 0 

示例:

当轴提供了参数,可以计算指定轴上的最小值索引。考虑以下示例。

import numpy as np
Arr = np.arange(12).reshape(3,4)

print("Array 为:")
print(Arr)

#沿轴最小值索引=0
print("\n沿轴最小值索引=0")
print(np.argmin(Arr, axis=0))

#沿轴最小值的索引=1
print("\n沿轴最小值索引=1")
print(np.argmin(Arr, axis=1)) 

上述代码的输出将是:

Array 为:
[[ 0  1  2  3]
 [ 4  5  6  7]
 [ 8  9 10 11]]

沿轴最小值索引=0
[0 0 0 0]

沿轴最小值索引=1
[0 0 0] 

示例:

当数组包含多个时最小值,函数返回第一次出现的索引。

import numpy as np
Arr = [20, 20, 10, 20, 30, 50, 10]

print("Array 为:", Arr)

#最小值索引
print("最小值指标为:", np.argmin(Arr)) 

上述代码的输出将是:

Array 为: [20, 20, 10, 20, 30, 50, 10]
最小值指标为: 2