python:完成代码,输入为numpy中的array一维数组nums,以列表格式返回该数组最大和最小值的位置(若存在多个最大/小值,则返回第一个)。 示例 1: 输入:array([28, 71, 52, 69, 87, 42]) 输出:[4, 0] 解析: 对于数组array([28, 71, 52, 69, 87, 42]),最大值的位置为4,最小值的位置为0,则返回[4, 0] 示例 2: 输入:array([76, 0, 40, 36, 25, 18, 98, 0]) 输出:[6, 1]
时间: 2024-03-05 11:53:38 浏览: 21
以下是给出的代码实现:
```python
import numpy as np
def find_max_min_index(nums):
max_index = np.argmax(nums)
min_index = np.argmin(nums)
return [max_index, min_index]
```
使用 `numpy.argmax` 和 `numpy.argmin` 函数可以找到数组中最大值和最小值的位置,然后将位置以列表的形式返回即可。
阅读全文