Python中的numpy.argmin()

NumPy的argmin()函数返回数组中每个轴的最小元素的索引。

语法

参数

• array : 输入数组，用于操作
• axis : [int, optional] 沿指定轴，如0或1
• out : [array optional] 提供将输出插入到输出数组的特性，它应该具有适当的形状和数据类型

返回值

此函数返回一个索引数组，该数组具有与输入数组相似的形状，并消除了轴长度维度。

代码 1

程序说明

此Python程序使用NumPy包生成一个介于0到15之间的4x4随机整数数组。然后使用argmin()函数来提供每列中最小项的索引。数组结构在代码注释中进行了说明，其中还显示了最小项以及它们在各列中的相应索引。输入数组和沿每列的最小项的计算出的索引在最终结果中显示。

输出

INPUT ARRAY:
 [[ 4  7 14 12]
 [ 1  5  8 14]
 [14  9  2  7]
 [ 0 13  0  8]]
Indices of the minimum element along columns: [3 1 3 2]

代码 2

程序说明

此Python程序说明了如何使用NumPy库的argmin()方法。第一步是使用NumPy的arange()和reshape()函数创建二维数组。然后对数组的特定项进行排序。在打印更新后的数组后，程序使用np.min()来查找并报告数组的最小元素。

此外，它使用np.argmin()和axis=0来查找并输出沿各列的最小项的第一次出现的所有索引。结果提供了关于最小项、更新后的数组以及与之对应的沿各列的索引的信息。

输出

array : 
 [[0 1 2 3 4]
 [5 6 7 8 9]]
array : 
 [[10  1  2  3  4]
 [ 5  1  7  8  9]]
min element:  1
min element indices along columns :  [1 0 0 0 0]