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 相关文章推荐
pyqt和pyside开发图形化界面
Jan 22 Python
python list转dict示例分享
Jan 28 Python
收藏整理的一些Python常用方法和技巧
May 18 Python
python实现矩阵乘法的方法
Jun 28 Python
Python2.X/Python3.X中urllib库区别讲解
Dec 19 Python
python如何对实例属性进行类型检查
Mar 20 Python
机器学习实战之knn算法pandas
Jun 22 Python
pyqt5 使用cv2 显示图片,摄像头的实例
Jun 27 Python
python实现登录密码重置简易操作代码
Aug 14 Python
关于Python 中的时间处理包datetime和arrow的方法详解
Mar 19 Python
详解pycharm配置python解释器的问题
Oct 15 Python
matplotlib对象拾取事件处理的实现
Jan 14 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
77A一级收信机修理记
2021/03/02 无线电
PHP转换IP地址到真实地址的方法详解
2013/06/09 PHP
解析:通过php socket并借助telnet实现简单的聊天程序
2013/06/18 PHP
php的mkdir()函数创建文件夹比较安全的权限设置方法
2014/07/28 PHP
PHP指定截取字符串中的中英文或数字字符的实例分享
2016/03/18 PHP
php抽象方法和抽象类实例分析
2016/12/07 PHP
jQuery 学习入门篇附实例代码
2010/03/16 Javascript
7款吸引人眼球的jQuery/CSS3特效实例分享
2013/04/25 Javascript
判断滚动条到底部的JS代码
2013/11/04 Javascript
JS实现选择TextArea内文本的方法
2015/08/03 Javascript
javascript实现状态栏中文字动态显示的方法
2015/10/20 Javascript
基于javascript实现图片滑动效果
2016/05/07 Javascript
卸载安装Node.js与npm过程详解
2016/08/15 Javascript
Validform表单验证总结篇
2016/10/31 Javascript
在点击div中的p时,如何阻止事件冒泡
2017/02/07 Javascript
Vue-Router模式和钩子的用法
2018/02/28 Javascript
JS实现的点击按钮图片上下滚动效果示例
2019/01/28 Javascript
通过npm或yarn自动生成vue组件的方法示例
2019/02/12 Javascript
JavaScript解析JSON数据示例
2019/07/16 Javascript
vue2 拖动排序 vuedraggable组件的实现
2019/08/08 Javascript
微信小程序绘制半圆(弧形)进度条
2020/11/18 Javascript
Python实现获取网站PR及百度权重
2015/01/21 Python
python读取文本中数据并转化为DataFrame的实例
2018/04/10 Python
python3 实现对图片进行局部切割的方法
2018/12/05 Python
python模拟点击网页按钮实现方法
2020/02/25 Python
Python 爬虫的原理
2020/07/30 Python
使用numpngw和matplotlib生成png动画的示例代码
2021/01/24 Python
欧舒丹英国官网:购买欧舒丹护手霜等明星产品
2017/01/17 全球购物
Beauty Expert美国/加拿大:购买奢侈美容产品
2018/12/05 全球购物
澳大利亚个性化儿童礼品网站:Bright Star Kids
2019/06/14 全球购物
写clone()方法时,通常都有一行代码,是什么?
2012/10/31 面试题
小学教师节活动方案
2014/01/31 职场文书
《庐山的云雾》教学反思
2014/04/22 职场文书
2016母亲节感恩话语
2015/12/09 职场文书
教师远程培训心得体会
2016/01/09 职场文书
MySQL索引失效的典型案例
2021/06/05 MySQL