numpy之argmax()函数

您所在的位置：网站首页 › python输出一组数中的最大值函数 › numpy之argmax()函数

numpy之argmax()函数

#numpy之argmax()函数| 来源: 网络整理| 查看: 265

numpy.argmax(a, axis=None, out=None)

此函数包括三个参数。

函数功能，返回最大值的索引；若axis=1，表明按行比较，输出每行中最大值的索引，若axis=0，则输出每列中最大值的索引。

举例说明

1 axis参数不出现时，此时将数组平铺，找出其中最大的那个值的索引。

一、最大值只有一个 import numpy as np a = np.array([[5, 7, 3, 2], [8, 6, 4, 9]]) np.argmax(a) 输出结果： 7 二、最大值有两个 import numpy as np a = np.array([[5, 7, 3, 2], [9, 6, 4, 9]]) np.argmax(a) 输出结果： 4 此时只会显示出现最早的那个最大值索引。

2 axis为1

import numpy as np a = np.array([[5, 7, 3, 2], [9, 6, 4, 9]]) np.argmax(a,axis=1) 输出结果： array([1, 0])

3 axis为0

import numpy as np a = np.array([[5, 7, 3, 2], [9, 6, 4, 9]]) np.argmax(a,axis=0) 输出结果： array([1, 0, 1, 1])

【本文地址】

numpy之argmax()函数

numpy之argmax()函数

今日新闻

推荐新闻