python topk()函数求最大和最小值实例


Posted in Python onApril 02, 2020

函数介绍

a.topk()求a中的最大值或最小值,返回两个值,一个是a中的值(最大或最小),一个是这个值的索引。

代码示例

>>> import torch
>>> a=torch.randn((3,5))
>>> a
tensor([[-0.4790, -0.6308, 0.2370, 0.0380, -0.0579],
  [-0.6712, -3.5483, -0.2370, -0.8658, 0.4145],
  [-1.4126, -0.8786, -0.4216, -0.0878, -1.4015]])
>>> _,pre=a.topk(1,dim=1,largest=True)
>>> pre
tensor([[2],
  [4],
  [3]])
>>> _
tensor([[ 0.2370],
  [ 0.4145],
  [-0.0878]])
>>> _,pre=a.topk(1,dim=1,largest=False)
>>> pre
tensor([[1],
  [1],
  [0]])
>>> _
tensor([[-0.6308],
  [-3.5483],
  [-1.4126]])

dim=1,为按行求最大最小值,largest为Ture,求最大值,largest=False,求最小值。

补充知识:python topN 取 最大的N个数 或 最小的N个数

来看代码吧!

import numpy as np
a = np.array([1,4,3,5,2])
b = np.argsort(a)
print(b)

print结果[0 4 2 1 3]

说明a[0]最小,a[3]最大

a[0]<a[4]<a[2]<a[1]<a[3]

以上这篇python topk()函数求最大和最小值实例就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持三水点靠木。

Python 相关文章推荐
python生成验证码图片代码分享
Jan 28 Python
python中文件变化监控示例(watchdog)
Oct 16 Python
快速了解Python中的装饰器
Jan 11 Python
python批量修改文件编码格式的方法
May 31 Python
Python列表list排列组合操作示例
Dec 18 Python
Python删除n行后的其他行方法
Jan 28 Python
详解Python爬取并下载《电影天堂》3千多部电影
Apr 26 Python
Python整数对象实现原理详解
Jul 01 Python
python Django里CSRF 对应策略详解
Aug 05 Python
Python迭代器协议及for循环工作机制详解
Jul 14 Python
Python代码注释规范代码实例解析
Aug 14 Python
python中pyplot基础图标函数整理
Nov 10 Python
python使用梯度下降和牛顿法寻找Rosenbrock函数最小值实例
Apr 02 #Python
Python如何通过百度翻译API实现翻译功能
Apr 02 #Python
Python基于百度AI实现OCR文字识别
Apr 02 #Python
python 穷举指定长度的密码例子
Apr 02 #Python
python3安装OCR识别库tesserocr过程图解
Apr 02 #Python
python简单的三元一次方程求解实例
Apr 02 #Python
Python 线性回归分析以及评价指标详解
Apr 02 #Python
You might like
PHP获取文件绝对路径的代码(上一级目录)
2011/05/29 PHP
php+MySQL判断update语句是否执行成功的方法
2014/08/28 PHP
将PHP程序中返回的JSON格式数据用gzip压缩输出的方法
2016/03/03 PHP
CI(CodeIgniter)框架视图中加载视图的方法
2017/03/24 PHP
PHP 中TP5 Request 请求对象的实例详解
2017/07/31 PHP
PHP 实现 WebSocket 协议原理与应用详解
2020/04/22 PHP
详解new function(){}和function(){}() 区别分析
2008/03/22 Javascript
Jquery 弹出层插件实现代码
2009/10/24 Javascript
js在指定位置增加节点函数insertBefore()用法实例
2015/01/12 Javascript
JQuery中节点遍历方法实例
2015/05/18 Javascript
jQuery 判断图片是否加载完成方法汇总
2015/08/10 Javascript
JS简单实现多级Select联动菜单效果代码
2015/09/06 Javascript
跟我学习javascript的浮点数精度
2015/11/16 Javascript
WEB前端实现裁剪上传图片功能
2016/10/17 Javascript
微信小程序 devtool隐藏的秘密
2017/01/21 Javascript
jquery获取链接地址和跳转详解(推荐)
2017/08/15 jQuery
解析vue data不可以使用箭头函数问题
2018/07/03 Javascript
详解微信小程序实现WebSocket心跳重连
2018/07/31 Javascript
axios携带cookie配置详解(axios+koa)
2018/12/28 Javascript
javascript判断一个变量是数组还是对象
2019/04/10 Javascript
vue 解决路由只变化参数页面组件不更新问题
2019/11/05 Javascript
python中定义结构体的方法
2013/03/04 Python
Python实现根据指定端口探测服务器/模块部署的方法
2014/08/25 Python
Python MySQL数据库连接池组件pymysqlpool详解
2017/07/07 Python
Python 多进程并发操作中进程池Pool的实例
2017/11/01 Python
详谈python3中用for循环删除列表中元素的坑
2018/04/19 Python
浅谈Python __init__.py的作用
2020/10/28 Python
删除pycharm鼠标右键快捷键打开项目的操作
2021/01/16 Python
绢花、人造花和人造花卉:BLOOM
2019/08/07 全球购物
室内设计实习自我鉴定
2013/09/25 职场文书
师范毕业生个人求职信
2013/12/09 职场文书
公司财务会计主管应聘求职信
2014/09/26 职场文书
2015年感恩节活动总结
2015/03/24 职场文书
医院志愿者活动总结
2015/05/06 职场文书
中学总务处工作总结
2015/08/12 职场文书
小学音乐课教学反思
2016/02/18 职场文书