np.argmax ()
时间: 2023-11-09 15:07:10 浏览: 31
`np.argmax()` 是一个 NumPy 库中的函数,用于返回一个数组中最大值的索引。该函数的语法为:`np.argmax(array, axis=None)`。其中,`array` 表示要寻找最大值的数组,`axis` 表示要在哪个轴上寻找最大值。如果不指定 `axis`,则默认返回所有元素中的最大值索引。例如,`np.argmax([1, 2, 3, 2, 1])` 返回 2,表示在给定的数组中,最大值 3 的索引是 2。
相关问题
np.argmax axis
np.argmax(axis)是一个NumPy函数,用于在指定轴上返回数组中最大元素的索引。它接受一个参数axis,用于指定要在哪个轴上进行比较。
例如,我们有一个二维数组arr:
```
arr = np.array([[1, 3, 2],
[4, 2, 9],
[7, 6, 5]])
```
如果我们想要在每一列中找到最大元素的索引,我们可以使用np.argmax(axis=0):
```
np.argmax(arr, axis=0)
```
输出将是一个包含每一列最大元素索引的一维数组:
```
array([2, 2, 1])
```
如果我们想要在每一行中找到最大元素的索引,我们可以使用np.argmax(axis=1):
```
np.argmax(arr, axis=1)
```
输出将是一个包含每一行最大元素索引的一维数组:
```
array([1, 2, 0])
```
请注意,索引是从0开始计数的。
python np.argmax
np.argmax 是 NumPy 库中的一个函数,用于返回数组中最大值的索引。它的语法如下:
np.argmax(array, axis=None, out=None)
其中,array 是要查找最大值的数组,axis 参数是可选的,指定沿着哪个轴进行查找。如果不指定 axis,则默认为 None,将数组平铺后进行查找。out 参数也是可选的,用于指定输出结果的位置。
例如,假设有一个二维数组 arr:
arr = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
要找到 arr 中的最大值的索引,可以使用 np.argmax:
max_index = np.argmax(arr)
这样,max_index 的值为 8,表示数组中最大值 9 所在的索引位置。