python topk()函數(shù)求最大和最小值實(shí)例
更新時(shí)間：2020年04月02日 10:00:13 作者：wang xiang
這篇文章主要介紹了python topk()函數(shù)求最大和最小值實(shí)例，具有很好的參考價(jià)值，希望對大家有所幫助。一起跟隨小編過來看看吧
函數(shù)介紹
a.topk()求a中的最大值或最小值，返回兩個(gè)值，一個(gè)是a中的值（最大或最?。?，一個(gè)是這個(gè)值的索引。
代碼示例
>>> 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，求最小值。
補(bǔ)充知識：python topN 取 最大的N個(gè)數(shù) 或 最小的N個(gè)數(shù)
來看代碼吧！
import numpy as np a = np.array([1,4,3,5,2]) b = np.argsort(a) print(b)
print結(jié)果[0 4 2 1 3]
說明a[0]最小，a[3]最大
a[0]<a[4]<a[2]<a[1]<a[3]
以上這篇python topk()函數(shù)求最大和最小值實(shí)例就是小編分享給大家的全部內(nèi)容了，希望能給大家一個(gè)參考，也希望大家多多支持腳本之家。
相關(guān)文章
Python簡單實(shí)現(xiàn)區(qū)域生長方式
今天小編就為大家分享一篇Python簡單實(shí)現(xiàn)區(qū)域生長方式，具有很好的參考價(jià)值，希望對大家有所幫助。一起跟隨小編過來看看吧2020-01-01
Pandas過濾dataframe中包含特定字符串的數(shù)據(jù)方法
今天小編就為大家分享一篇Pandas過濾dataframe中包含特定字符串的數(shù)據(jù)方法，具有很好的參考價(jià)值，希望對大家有所幫助。一起跟隨小編過來看看吧2018-11-11
今天小編就為大家分享一篇tensorflow 查看梯度方式，具有很好的參考價(jià)值，希望對大家有所幫助。一起跟隨小編過來看看吧2020-02-02
這篇文章主要介紹了Python如何獲取多線程返回結(jié)果，具有很好的參考價(jià)值，希望對大家有所幫助。如有錯(cuò)誤或未考慮完全的地方，望不吝賜教2022-05-05
這篇文章主要介紹了PyQt5 QListView 高亮顯示某一條目的案例，具有很好的參考價(jià)值，希望對大家有所幫助。一起跟隨小編過來看看吧2021-03-03
在本篇內(nèi)容中我們給大家總結(jié)了關(guān)于Python3標(biāo)準(zhǔn)庫的相關(guān)內(nèi)容，需要的朋友們跟著學(xué)習(xí)下。2019-02-02
使用Python實(shí)現(xiàn)視頻封面批量下載器
在視頻網(wǎng)站上,每個(gè)視頻都有一個(gè)獨(dú)特的封面圖像,本文主要為大家詳細(xì)如何使用Python編寫一個(gè)視頻封面批量下載器,感興趣的小伙伴可以跟隨小編一起學(xué)習(xí)一下2024-04-04
這篇文章主要為大家詳細(xì)介紹了python爬取足球直播吧五大聯(lián)賽積分榜，具有一定的參考價(jià)值，感興趣的小伙伴們可以參考一下2018-06-06
這篇文章主要介紹了將python程序打包成DLL的三種方式，具有很好的參考價(jià)值，希望對大家有所幫助。如有錯(cuò)誤或未考慮完全的地方，望不吝賜教2023-02-02
這篇文章主要為大家詳細(xì)介紹了python如何將兩張圖片生成為全景圖片，文中示例代碼介紹的非常詳細(xì)，具有一定的參考價(jià)值，感興趣的小伙伴們可以參考一下2020-03-03