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 线程池threadpool之实现
Nov 17 Python
Python中对象的引用与复制代码示例
Dec 04 Python
Python基于分析Ajax请求实现抓取今日头条街拍图集功能示例
Jul 19 Python
Python matplotlib通过plt.scatter画空心圆标记出特定的点方法
Dec 13 Python
python内置函数sorted()用法深入分析
Oct 08 Python
Python程序暂停的正常处理方法
Nov 07 Python
django连接mysql数据库及建表操作实例详解
Dec 10 Python
Python timeit模块的使用实践
Jan 13 Python
Python执行时间的几种计算方法
Jul 31 Python
Python连接mysql方法及常用参数
Sep 01 Python
Flask-SocketIO服务端安装及使用代码示例
Nov 26 Python
python神经网络学习 使用Keras进行简单分类
May 04 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 正则表达式小结
2009/08/31 PHP
PHP 金额数字转换成英文
2010/05/06 PHP
成为好程序员必须避免的5个坏习惯
2014/07/04 PHP
phalcon model在插入或更新时会自动验证非空字段的解决办法
2016/12/29 PHP
PHP 年月日的三级联动实例代码
2017/05/24 PHP
PHP实现SMTP邮件的发送实例
2018/09/27 PHP
PHP如何通过date() 函数格式化显示时间
2020/11/13 PHP
理解JavaScript的caller,callee,call,apply
2009/04/28 Javascript
jQuery效果 slideToggle() 方法(在隐藏和显示之间切换)
2011/06/28 Javascript
jquery数据验证插件(自制,简单,练手)实例代码
2013/10/24 Javascript
javascript原生和jquery库实现iframe自适应高度和宽度
2014/07/18 Javascript
JS中的form.submit()不能提交表单的错误原因
2014/10/08 Javascript
轻松创建nodejs服务器(3):代码模块化
2014/12/18 NodeJs
基于JavaScript将表单序列化类型的数据转化成对象的处理(允许对象中包含对象)
2015/12/28 Javascript
Vue中计算属性computed的示例解读
2017/07/26 Javascript
vue slots 组件的组合/分发实例
2018/09/06 Javascript
[01:39](回顾)各路豪强针锋相对,几经鏖战四强产生
2014/07/01 DOTA
Python中threading模块join函数用法实例分析
2015/06/04 Python
Python中的字符串类型基本知识学习教程
2016/02/04 Python
Python实现按当前日期(年、月、日)创建多级目录的方法
2018/04/26 Python
Python堆排序原理与实现方法详解
2018/05/11 Python
Python3匿名函数用法示例
2018/07/25 Python
详解使用PyInstaller将Pygame库编写的小游戏程序打包为exe文件
2019/08/23 Python
python 微信好友特征数据分析及可视化
2020/01/07 Python
详解python中GPU版本的opencv常用方法介绍
2020/07/24 Python
Python爬虫爬取糗事百科段子实例分享
2020/07/31 Python
使用python操作lmdb对数据读取的实例
2020/12/11 Python
Java面试笔试题大全
2016/11/23 面试题
计算机科学系职业生涯规划书
2014/03/08 职场文书
如何写一份好的英文求职信
2014/03/19 职场文书
土木工程专业本科生求职信
2014/10/01 职场文书
2015年小学开学寄语
2015/02/27 职场文书
关于开学的感想
2015/08/10 职场文书
竞聘开场白方式有哪些?
2019/08/28 职场文书
java executor包参数处理功能 
2022/02/15 Java/Android
Smart 2 车辆代号 HC11 全新谍照曝光
2022/04/21 数码科技