详解Numpy argmin()（返回数组元素的最小值的索引）函数的作用与使用方法

Numpy argmin()函数用于返回数组中指定轴上最小值的索引。在本攻略中，我将提供argmin()函数的使用方法、语法和参数设置，并展示两个实例来说明如何使用该函数。

函数的语法和参数设置

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

参数说明：

• a：数组。
• axis：用于计算最小值的轴。如果未指定，则所有元素被视为单个数组，并且返回整个数组中最小值的索引。
• out：输出数组，用于保存argmin()函数的结果。

使用方法

我们可以将argmin()函数与其他numpy函数一起使用，例如numpy.array()和numpy.linspace()，首先定义一个数组或生成一个数组。然后，使用argmin()函数来计算最小值的索引。

例子一

下面是一个简单的示例，展示如何使用argmin()函数来计算数组中的最小值索引

import numpy as np

a = np.array([[3,7,5],[8,4,3],[2,4,9]])

print('原始数组:\n', a)

print('最小值索引:', np.argmin(a))

print('沿轴0的最小值索引:', np.argmin(a, axis= 0))

print('沿轴1的最小值索引:', np.argmin(a, axis = 1))

输出结果：

原始数组:
 [[3 7 5]
 [8 4 3]
 [2 4 9]]
最小值索引: 6
沿轴0的最小值索引: [2 1 1]
沿轴1的最小值索引: [0 2 0]

例子二

下面是另一个示例，展示如何在numpy.linspace()函数的范围内计算最小值的索引

import numpy as np

a = np.linspace(0,9,10)

print('原始数组:\n', a)

print('最小值索引:', np.argmin(a))

print('沿第0轴的最小值索引:', np.argmin(a, axis= 0))

输出如下：

原始数组:
 [0. 1. 2. 3. 4. 5. 6. 7. 8. 9.]
最小值索引: 0
沿第0轴的最小值索引: 0

在第二个示例中，我没有指定轴，因此numpy.argmin()默认计算整个数组的最小值索引，因为数组a只包含一个轴，所以结果与不指定轴的结果相同。在第一个示例中，我将指定轴axis = 0和axis = 1来计算数组a的最小值索引。

总结

numpy.argmin()函数是一个非常实用的函数，它用于返回数组中指定轴上最小值的索引。它有一个axis参数，可以让我们指定轴，最小值索引计算就会相对于该轴进行。对于处理和分析numpy数组的工作，argmin()函数是一个非常重要的函数之一。