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