numpy ndarray 取出滿足特定條件的某些行實(shí)例
在進(jìn)行物體檢測(cè)的ground truth boxes annotations包圍框坐標(biāo)數(shù)據(jù)整理時(shí),需要實(shí)現(xiàn)這樣的功能:
numpy里面,對(duì)于N*4的數(shù)組,要實(shí)現(xiàn)對(duì)于每一行,如果第3列和第1列數(shù)值相等或者第2列和第0列數(shù)值相等,就刪除這一行,要返回保留下來(lái)的numpy數(shù)組 shape M*4
對(duì)于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對(duì)象,在豎直方向上進(jìn)行數(shù)組拼接
其中l(wèi)ist或者tuple中的每個(gè)元素是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í)例就是小編分享給大家的全部?jī)?nèi)容了,希望能給大家一個(gè)參考,也希望大家多多支持腳本之家。
相關(guān)文章
Python實(shí)現(xiàn)郵件自動(dòng)下載的示例詳解
這篇文章主要為大家詳細(xì)介紹了如何利用Python語(yǔ)言實(shí)現(xiàn)郵件自動(dòng)下載以及附件解析功能,文中的示例代碼講解詳細(xì),感興趣的小伙伴可以了解一下2022-07-07
Macbook air m1安裝python/anaconda全過(guò)程(圖文)
這篇文章主要介紹了Macbook air m1安裝python/anaconda全過(guò)程(圖文),文中通過(guò)圖文介紹的非常詳細(xì),對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值,需要的朋友們下面隨著小編來(lái)一起學(xué)習(xí)學(xué)習(xí)吧2021-03-03
Python使用Rich?type和TinyDB構(gòu)建聯(lián)系人通訊錄
這篇文章主要為大家介紹了Python使用Rich?type和TinyDB構(gòu)建聯(lián)系人通訊錄應(yīng)用程序,有需要的朋友可以借鑒參考下,希望能夠有所幫助,祝大家多多進(jìn)步,早日升職加薪2022-08-08
python中根據(jù)字符串調(diào)用函數(shù)的實(shí)現(xiàn)方法
下面小編就為大家?guī)?lái)一篇python中根據(jù)字符串調(diào)用函數(shù)的實(shí)現(xiàn)方法。小編覺(jué)得挺不錯(cuò)的,現(xiàn)在就分享給大家,也給大家做個(gè)參考,一起跟隨小編過(guò)來(lái)看看吧2016-06-06
Python使用type動(dòng)態(tài)創(chuàng)建類操作示例
這篇文章主要介紹了Python使用type動(dòng)態(tài)創(chuàng)建類操作,結(jié)合實(shí)例形式詳細(xì)分析了Python使用type動(dòng)態(tài)創(chuàng)建類的具體原理、實(shí)現(xiàn)方法與操作注意事項(xiàng),需要的朋友可以參考下2020-02-02

