Python小技巧 - argmax

最新推荐文章于 2025-04-03 14:25:48 发布
最新推荐文章于 2025-04-03 14:25:48 发布
阅读量2w 收藏 29
点赞数 4
分类专栏： Python
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.csdn.net/byron123456sfsfsfa/article/details/88923085
版权

argmax 返回的是输入列表中最大值的位置，其重要性不必多言，但是据我所知 Python 自带的库中只提供 max 这个函数，并没有 argmax，但是实现起来不难。

Numpy 中的 argmax

首先我们先来看一下 Numpy 中提供 argmax 函数，它重要的特点就是在有多个最大值的情况下，只返回第一个出现的最大值的位置。

In [1]: import numpy as np
In [2]: a = [1, 2, 9, 2, 5, 6, 9]
In [3]: np.argmax(a)
Out[3]: 2

如果需要返回所有最大值的位置的话，还是要麻烦一下的：

In
最低0.47元/天 解锁文章
python数据分析numpy基础之argmax求数组最大值索引
sinat_34735632的博客
03-02 2447
python的numpy库的argmax()函数，用于获取沿指定轴的最大值的索引。argmax()返回沿指定轴的最大值的索引。入参axis表示指定轴，默认为None表示整个数组的最大值的索引而不是某个轴的最大值索引，即转为一维数组后的最大值的索引。axis只能为整数，不支持整数元组。入参keepdims表示是否保留指定轴的尺寸为1，默认为False，不保留。
python笔记argmax()
qq_44722189的博客
12-07 874
参考：https://blog.csdn.net/nyist_yangguang/article/details/118521240。axis = 0，简单理解，返回结果为：数组中每一列最大值所在“行”索引值。axis = 1，简单理解，返回结果为：数组中每一行最大值所在“列”索引值。参数axis可取(0,1)，默认是0，表示数组第几维的最大值。返回数组中最大值元素的索引位置。
PyTorch 中的一个函数 —— torch.argmax
最新发布
HGX-DJK
04-03 504
​输入:input：输入张量。dim（可选）：指定沿哪个维度查找最大值。如果为 None，则在整个张量中查找。keepdim（可选）：是否保持输出张量的维度与输入一致（默认为 False）。​输出：一个张量，包含最大值所在的索引torch.argmax 是一个高效的工具，广泛应用于分类模型预测、指标计算等场景。理解其 dim 和 keepdim 参数的行为，可以灵活处理不同维度的数据。
python之argmax
12-13 2594
argmax 可以返回最大值得索引值，举例如下： import numpy as np a = np.array([3, 1, 2, 4, 6, 1]) print(np.argmax(a)) 这里最大的值是6，索引从0开始，所以输出应该是4。 运行结果如下：  果然是4....
python——argmax函数
生活如此美好，致敬每一个热爱“热爱”的人！
02-27 4366
def argmax(a, axis=None, out=None, *, keepdims=np._NoValue): """ Returns the indices of the maximum values along an axis. Parameters ---------- a : array_like Input array. axis : int, optional By default, the inde..
python中numpy函数：argmax
YNJ的博客
01-23 1520
Nmupy 选取最大值的函数：argmax与amax argmax和amax两个函数都是与最大值相关的函数，这两这有什么区别呢？ argmax返回的是那个值本身，amax返回的是那个最大值的标签 范例如下： import numpy as np a=np.array([[1,2,3],[4,5,6],[7,8,9]]) print(a) amax_=np.amax(a,axis=1) print('amax_ : ') print(amax_) argmax_=np.argmax(a,axis=1) pri
python基础
haonanxu的博客
04-08 930
点击这里，边看视频讲解，边学习以下内容有的函数执行一个流程就完了，并不需要返回什么信息，比如print函数，就是打印一下参数字符串，就完成了它的任务。但是有的函数，需要返回一个结果给调用者。比如 计算两个数字平方和的函数，必须要返回一个结果，告诉调用代码，平方和的计算结果。函数要返回结果，就必须要使用return关键字，后面加上返回的结果对象print(ret)语句等号右边的 就是函数调用的表达式。
Python学习笔记-基础篇
cy2lyf的博客
09-07 3220
python学习笔记-基础篇
整理了100个Python小技巧（超级实用）
python2021_的博客
04-20 3068
目前Python可以说是非常流行，在目前的编程语言中，Python的抽象程度是最高的，是最接近自然语言的，很容易上手。你可以用它来完成很多任务，比如数据科学、机器学习、Web开发、脚本编写、自动化等。 下面，我就给大家分享100个Python小技巧，帮助大家更好的了解和学习Python，欢迎收藏、关注，点赞支持！ ▍1、for循环中的else条件 这是一个for-else方法，循环遍历列表时使用else语句。下面举个例子，比如我们想检查一个列表中是否包含奇数。那么可以通过for循环，遍历查找。 n
性能提升秘籍：Python优化IEC-61850通信性能的黄金法则（性能秘籍）
本文首先介绍了IEC-61850通信协议的基础知识，并着重探讨了Python语言在IEC-61850通信中的应用，包括如何选择合适的Python库以实现客户端和服务器端的功能。随后，本文深入分析了Python在IEC-61850性能优化方面
Python:NumPy教程-numpy.argmax()在Python中的使用
白帽阿叁的博客
10-24 2447
在许多情况下，数组的大小过大时，从中找出最大元素需要太长时间。为此，Python的numpy模块提供了一个名为numpy.argmax()的函数。该函数返回最大值的索引以及指定的轴。
【Python】Numpy中的argmax()函数
weixin_45057858的博客
08-13 8091
【Python】Numpy中的argmax()函数
学习记录_python之argmax
B12345r的博客
05-22 321
import numpy as np x=np.array([[1,2,3],[4,5,9],[9,9,9]]) y1=np.argmax(x) #表示在所有元素寻找最大值，只返回第一次最大值的位置 y2=np.argmax(x,axis=0) #axis=0 表示沿着列寻找最大值 y3=np.argmax(x,axis=1) #axis=1 表示沿着行寻找最大值 print(y1) print(y2) print(y3) ...
Python学习笔记6：Numpy.argmax()
Softdiamonds的博客
05-05 616
用于返回的是x所在的最大值的下标的值。>>> a = np.arange(6).reshape(2,3) >>> a array([[0, 1, 2], [3, 4, 5]]) #默认是返回的所有的元素的最大的一个下标 >>> np.argmax(a) 5 ＃axis = 0 ,表示以行为基准，比较在相同列上的...
Python基本函数:np.argmax（
OneBaBa的专栏
10-20 1万+
Python基本函数:np.argmax（） 一、函数说明 二、函数用法 格式：np.argmax(a)注意：返回的是a中元素最大值所对应的索引值 一、函数说明 <span style="color:#000000"><code class="language-python">argmax<span style="color:#999999">(</span>a<span style="color:#999999">,&l..
NumPy之argmax()方法理解
热门推荐
zhaoguanhua的博客
07-20 1万+
python numpy之argmax理解
(Python)对numpy中的argmax用法的理解
qq_52363625的博客
01-18 6282
作用 argmax()返回的就是最大数的索引 argmax()有一个参数axis，可以指定函数返回不同维的最大值。 一维 import numpy as np a = np.array([1, 2, 3, 4, 5]) print(np.argmax(a, axis=0)) 结果是4。而在python中一维向量只有一个方向，所以一维情况下axis只能等于0，只会返回一个值。 如果对这个它进行转置，它就会变成一个二维矩阵： a = np.array([1, 3, 5, 7]) print(a) p
numpy的argmax用法（python）
u012413709的博客
06-02 465
https://www.cnblogs.com/touch-skyer/p/8509217.html
torch,argmax
04-01
### PyTorch 中 `argmax` 函数的用法 在 PyTorch 中，`torch.argmax(input, dim=None)` 是一个用于返回输入张量沿指定维度的最大值索引的函数。如果未提供 `dim` 参数，则会将整个张量展平并计算全局最大值的索引[^1]。 以下是关于该函数的一些重要特性及其使用方法： #### 基本语法 ```python torch.argmax(input, dim=None, keepdim=False) ``` - **参数说明**： - `input`: 输入张量。 - `dim`: 指定沿着哪个维度查找最大值的索引。如果不设置此参数，默认会在整个张量上操作，并将其视为一维向量。 - `keepdim`: 如果设为 `True`，则输出张量将在指定维度保留大小为 1 的形状；否则，该维度会被压缩掉。 #### 使用示例 ##### 示例 1: 整体展开求解最大值索引 当不指定 `dim` 参数时，张量会被视作一维数组来寻找最大值的位置。 ```python import torch tensor = torch.tensor([[1, 2], [3, 4]]) result = torch.argmax(tensor) # 返回整体中的最大值位置 (即 '4' 所处的位置) print(result) # 输出: tensor(3), 表明按列优先顺序计数得到的结果 ``` ##### 示例 2: 沿特定维度获取最大值索引 通过设定 `dim` 参数可以控制在哪一维度上执行操作。 ```python row_max_indices = torch.argmax(tensor, dim=1) # 对每一行分别找最大值所在列号 print(row_max_indices) # 输出: tensor([1, 1]) column_max_indices = torch.argmax(tensor, dim=0) # 对每列分别找最大值所在的行号 print(column_max_indices) # 输出: tensor([1, 1]) ``` #### 应用场景 `torch.argmax()` 经常被应用于分类模型预测结果处理中，用来找出概率分布中最可能对应的类别标签。例如，在神经网络最后经过 softmax 层之后获得的概率矩阵里提取最终决策依据。 --- ### 结合其他功能扩展讨论 虽然上述介绍了基本用法，但在实际项目开发过程中可能会遇到更复杂的需求情况。比如结合之前提到过的 `torch.unbind()[^2]` 或者 Edward library 下面涉及到变分推断(Variational Inference)时候使用的技巧[^3]等都可以进一步增强对于数据结构的理解以及优化算法实现效率等方面起到积极作用。 #### 小结代码片段展示如何综合运用多个工具完成任务目标如下所示: ```python # 创建模拟数据集 data_tensor = torch.randn((5, 3)) probs = torch.softmax(data_tensor, dim=-1) # 获取最高可能性类别的编号列表 predicted_classes = torch.argmax(probs, dim=1) print(predicted_classes) # 利用 unbind 方法分离批次样本单独分析 individual_samples = torch.unbind(data_tensor, dim=0) for idx, sample in enumerate(individual_samples): print(f'Sample {idx}:', torch.argmax(sample)) ```
评论
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

当前余额3.43元 前往充值 >
需支付：10.00元
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付元
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明：

1.余额是钱包充值的虚拟货币，按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载，可以购买VIP、付费专栏及课程。

余额充值