numpy ndarray 取出滿足特定條件的某些行實(shí)例
在進(jìn)行物體檢測的ground truth boxes annotations包圍框坐標(biāo)數(shù)據(jù)整理時，需要實(shí)現(xiàn)這樣的功能：
numpy里面，對于N*4的數(shù)組，要實(shí)現(xiàn)對于每一行，如果第3列和第1列數(shù)值相等或者第2列和第0列數(shù)值相等，就刪除這一行，要返回保留下來的numpy數(shù)組 shape M*4
對于numpy數(shù)組的操作要盡量避免for循環(huán)，因?yàn)閚umpy數(shù)組支持布爾索引。
import numpy as np a1=np.array( [1,0,1,5] ) a2=np.array( [0,8,5,8] ) center=np.random.randint(0,10,size=(3,4)) # print(a1.shape,a2.shape,center.shape) b=np.vstack((a1,center,a2)) '''
numpy vstack 所輸入的參數(shù)必須是list或者tuple的iterable對象，在豎直方向上進(jìn)行數(shù)組拼接
其中l(wèi)ist或者tuple中的每個元素是numpy.ndarray類型
它們必須具有相同的列數(shù)，拼接完成后行數(shù)增加
numpy hstack 在水平方向上進(jìn)行數(shù)組拼接
進(jìn)行拼接的數(shù)組必須具有相同的行數(shù)，拼接完成后列數(shù)增加
''' print(b.shape,b) out=b[b[:,3]!=b[:,1]] out2=out[out[:,2]!=out[:,0]] print(out2.shape,out2) ''' (5, 4) [[1 0 1 5] [6 9 9 1] [9 1 6 5] [2 8 8 1] [0 8 5 8]] (3, 4) [[6 9 9 1] [9 1 6 5] [2 8 8 1]] ''' b1=a1.reshape(-1,1) b2=a2.reshape(-1,1) before_list=[] before_list.append(b1) before_list.append(center.reshape(4,3)) before_list.append(b2) out3=np.hstack(before_list) print(out3.shape)#(4, 5)
以上這篇numpy ndarray 取出滿足特定條件的某些行實(shí)例就是小編分享給大家的全部內(nèi)容了，希望能給大家一個參考，也希望大家多多支持腳本之家。
相關(guān)文章
今天給大家分享的是如何使用Django實(shí)現(xiàn)簡單分頁功能的方法，主要依托于Django提供的一個新的類Paginator來實(shí)現(xiàn)的2018-07-07
這篇文章主要介紹了Python中with語句的用法,with語句的使用是Python學(xué)習(xí)過程當(dāng)中的基礎(chǔ)知識,本文來自于IBM官方技術(shù)文檔,需要的朋友可以參考下2015-04-04
Python卷積神經(jīng)網(wǎng)絡(luò)圖片分類框架詳解分析
在機(jī)器視覺領(lǐng)域中,卷積神經(jīng)網(wǎng)絡(luò)算法作為一種新興算法出現(xiàn),在圖像識別領(lǐng)域中,卷積神經(jīng)網(wǎng)絡(luò)能夠較好的實(shí)現(xiàn)圖像的分類效果,而且其位移和形變具有較高的容忍能力2021-11-11
對numpy的array和python中自帶的list之間相互轉(zhuǎn)化詳解
下面小編就為大家分享一篇對numpy的array和python中自帶的list之間相互轉(zhuǎn)化詳解，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧2018-04-04
這篇文章主要介紹了Python 函數(shù)返回值的示例代碼，文中通過示例代碼介紹的非常詳細(xì)，對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值，需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧2019-03-03
python利用PyQt5設(shè)計(jì)鼠標(biāo)顯示形狀
不知道大家有沒有發(fā)現(xiàn),我們在網(wǎng)頁移動鼠標(biāo)時,不同的網(wǎng)頁會有不同的鼠標(biāo)移動特效,通過移動鼠標(biāo),會形成類似蜘蛛網(wǎng)等等的特效,本文將用PyQt5實(shí)現(xiàn)這一特效,需要的可以參考一下2024-07-07
Python基于字典實(shí)現(xiàn)switch case函數(shù)調(diào)用
這篇文章主要介紹了Python基于字典實(shí)現(xiàn)switch case函數(shù)調(diào)用,文中通過示例代碼介紹的非常詳細(xì)，對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值,需要的朋友可以參考下2020-07-07
在本篇內(nèi)容中小編給大家分享的是關(guān)于python是否屬于腳本語言的相關(guān)內(nèi)容知識點(diǎn)，有興趣的朋友們可以參考下。2020-06-06
Python實(shí)現(xiàn)批量將符合要求的文件自動復(fù)制到新文件夾
這篇文章主要為大家詳細(xì)介紹了如何使用Python實(shí)現(xiàn)批量將文件名稱符合要求的文件自動復(fù)制到新文件夾,文中的示例代碼講解詳細(xì),有需要的小伙伴可以參考下2023-10-10
Python?Pandas:DataFrame一列切分成多列、分隔符切分選字段方式
這篇文章主要介紹了Python?Pandas:DataFrame一列切分成多列、分隔符切分選字段方式,具有很好的參考價值,希望對大家有所幫助,如有錯誤或未考慮完全的地方,望不吝賜教2023-09-09