python求列表中最小元素的索引,用numpy库的argmin()方法
时间: 2023-04-04 20:04:57 浏览: 138
可以使用numpy库的argmin()方法来求列表中最小元素的索引,具体代码如下:
import numpy as np
lst = [3, 5, 1, 7, 2, 9, 4]
index = np.argmin(lst)
print(index)
输出结果为:2
即列表中最小元素1的索引为2。
相关问题
python求列表中最小元素的索引的三种方法
可以使用以下三种方法来求列表中最小元素的索引:
1. 使用min()函数和index()方法:先使用min()函数找到列表中的最小值,然后使用index()方法找到该值在列表中的索引。
2. 使用for循环和enumerate()函数:使用for循环遍历列表中的每个元素,并使用enumerate()函数获取每个元素的索引和值,然后比较每个值,找到最小值的索引。
3. 使用numpy库的argmin()方法:将列表转换为numpy数组,然后使用argmin()方法找到最小值的索引。
numpy.argmax() 和 numpy.argmin()
numpy.argmax() 函数返回数组中最大值的索引,而 numpy.argmin() 函数返回数组中最小值的索引。这两个函数都可以接受一个参数 axis,用于指定沿着哪个轴计算最大值或最小值。如果不指定 axis 参数,则会将整个数组视为一个一维数组来计算最大值或最小值。例如:
```python
import numpy as np
a = np.array([[1, 2, 3], [4, 5, 1]])
print(np.argmax(a)) # 输出:4,即 a[1, 1] 的索引
print(np.argmax(a, axis=0)) # 输出:[1 1 0],即每列最大值的索引
print(np.argmin(a)) # 输出:0,即 a[0, 0] 的索引
print(np.argmin(a, axis=1)) # 输出:[0 2],即每行最小值的索引
```
阅读全文