numpy ndarray 取出滿足特定條件的某些行實例
更新時間：2019年12月05日 13:42:14 作者：WYXHAHAHA123
今天小編就為大家分享一篇numpy ndarray 取出滿足特定條件的某些行實例，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧
在進行物體檢測的ground truth boxes annotations包圍框坐標數(shù)據(jù)整理時，需要實現(xiàn)這樣的功能：
numpy里面，對于N*4的數(shù)組，要實現(xiàn)對于每一行，如果第3列和第1列數(shù)值相等或者第2列和第0列數(shù)值相等，就刪除這一行，要返回保留下來的numpy數(shù)組 shape M*4
對于numpy數(shù)組的操作要盡量避免for循環(huán)，因為numpy數(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對象，在豎直方向上進行數(shù)組拼接
其中l(wèi)ist或者tuple中的每個元素是numpy.ndarray類型
它們必須具有相同的列數(shù)，拼接完成后行數(shù)增加
numpy hstack 在水平方向上進行數(shù)組拼接
進行拼接的數(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 取出滿足特定條件的某些行實例就是小編分享給大家的全部內(nèi)容了，希望能給大家一個參考，也希望大家多多支持腳本之家。
相關(guān)文章
集合也也也也是python內(nèi)置的一種數(shù)據(jù)結(jié)構(gòu)，它是一個無序且元素不重復的序列。這里有兩個關(guān)鍵詞一個是無序，這一點和字典是一樣的，另一個關(guān)鍵詞是元素不重復，這一點和字典的key(鍵)是一樣的2020-02-02
本文主要介紹了Python中assert函數(shù)的使用(含源代碼)，文中通過示例代碼介紹的非常詳細，對大家的學習或者工作具有一定的參考學習價值，需要的朋友們下面隨著小編來一起學習學習吧2023-04-04
這篇文章主要為大家詳細介紹了如何利用Python語言實現(xiàn)郵件自動下載以及附件解析功能，文中的示例代碼講解詳細，感興趣的小伙伴可以了解一下2022-07-07
Macbook air m1安裝python/anaconda全過程(圖文)
這篇文章主要介紹了Macbook air m1安裝python/anaconda全過程(圖文)，文中通過圖文介紹的非常詳細，對大家的學習或者工作具有一定的參考學習價值，需要的朋友們下面隨著小編來一起學習學習吧2021-03-03
椒鹽噪聲是數(shù)字圖像中的常見噪聲,一般是圖像傳感器、傳輸信道及解碼處理等產(chǎn)生的黑白相間的亮暗點噪聲,椒鹽噪聲常由圖像切割產(chǎn)生,這篇文章主要給大家介紹了關(guān)于利用OpenCV給彩色圖像添加椒鹽噪聲的相關(guān)資料,需要的朋友可以參考下2021-10-10
在本篇文章中我們給大家介紹了關(guān)于django中使用定時任務(wù)的方法的相關(guān)知識點，有需要的朋友們參考下。2018-09-09
pyetl是一個純python開發(fā)的ETL框架， 相比sqoop, datax 之類的ETL工具，pyetl可以對每個字段添加udf函數(shù)，使得數(shù)據(jù)轉(zhuǎn)換過程更加靈活，相比專業(yè)ETL工具pyetl更輕量，純python代碼操作，更加符合開發(fā)人員習慣。這篇文章主要介紹了python ETL工具 pyetl,需要的朋友參考下2020-06-06
Python使用Rich?type和TinyDB構(gòu)建聯(lián)系人通訊錄
這篇文章主要為大家介紹了Python使用Rich?type和TinyDB構(gòu)建聯(lián)系人通訊錄應(yīng)用程序，有需要的朋友可以借鑒參考下，希望能夠有所幫助，祝大家多多進步，早日升職加薪2022-08-08
python中根據(jù)字符串調(diào)用函數(shù)的實現(xiàn)方法
下面小編就為大家?guī)硪黄猵ython中根據(jù)字符串調(diào)用函數(shù)的實現(xiàn)方法。小編覺得挺不錯的，現(xiàn)在就分享給大家，也給大家做個參考，一起跟隨小編過來看看吧2016-06-06
Python使用type動態(tài)創(chuàng)建類操作示例
這篇文章主要介紹了Python使用type動態(tài)創(chuàng)建類操作,結(jié)合實例形式詳細分析了Python使用type動態(tài)創(chuàng)建類的具體原理、實現(xiàn)方法與操作注意事項,需要的朋友可以參考下2020-02-02