Numpy argmin()函数用于返回数组中指定轴上最小值的索引。在本攻略中,我将提供argmin()函数的使用方法、语法和参数设置,并展示两个实例来说明如何使用该函数。
numpy.argmin(a, axis=None, out=None)
参数说明:
我们可以将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()函数是一个非常重要的函数之一。
本文链接:http://task.lmcjl.com/news/17113.html