python argmax，max，maximum用法

桃花巷低调海棠

于 2021-02-07 09:10:08 发布

阅读量1.9k

点赞数 2

CC 4.0 BY-SA版权

分类专栏： Python语言文章标签： python

本文链接：https://blog.csdn.net/jayqilixiang/article/details/113735151

Python语言专栏收录该内容

3 篇文章

订阅专栏

本文介绍了argmax和max在函数中的不同作用，argmax返回函数取得最大值时对应的参数，而max则返回最大值本身。通过Python numpy库的示例，展示了argmax和max以及maximum函数的应用，帮助读者深入理解这些概念。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

写在前面

偶遇argmax()函数，整理一下其用法。

argmax与max的区别

例如函数：y=f(t)

y=max f(t) ：y是f(t)函数的最大值

y=argmax f(t) ：y是f(t)函数取到最大值时的参数t

import numpy as np
a = np.array([[1, 5, 5, 2],
              [9, 6, 2, 8],
              [3, 7, 9, 1]])
print(np.argmax(a, axis=0))
# out: [1,2,2,1]

max, argmax, maximum对比

a = [1, 4, 5, 7, 2]
a1 = np.max(a)
a2 = np.argmax(a)
a3 = np.maximum(a, [1, 2, 3, 6, 9])  # 返回对应位置较大的元素值
a1, a2, a3
# out: 
(7, 3, array([1, 4, 5, 7, 9]))