如何获得一维numpy.ndarray中最大元素的位置
时间: 2024-05-01 10:23:54 浏览: 83
numpy-numpy教程之NDArray+Broadcasting+Indexing.zip
可以使用numpy库中的argmax()函数来获取一维numpy数组中最大元素的位置。argmax()函数返回的是最大元素的索引,如果数组中有多个最大元素,则返回第一个最大元素的位置。
以下是示例代码:
```python
import numpy as np
arr = np.array([1, 3, 2, 4, 2])
max_index = np.argmax(arr)
print("最大元素的位置是:", max_index)
```
输出结果为:
```
最大元素的位置是: 3
```
在上述代码中,我们首先创建了一个一维numpy数组`arr`,然后使用argmax()函数来获取最大元素的位置,并将其存储在变量`max_index`中。最后打印出`max_index`的值,即为最大元素的位置。
阅读全文