numpy ndarray 取出滿(mǎn)足特定條件的某些行實(shí)例

發(fā)布時(shí)間：2020-09-20 21:00:23 來(lái)源：腳本之家閱讀：232 作者：WYXHAHAHA123 欄目：開(kāi)發(fā)技術(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類(lèi)型

它們必須具有相同的列數(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 取出滿(mǎn)足特定條件的某些行實(shí)例就是小編分享給大家的全部?jī)?nèi)容了，希望能給大家一個(gè)參考，也希望大家多多支持億速云。

向AI問(wèn)一下細(xì)節(jié)

numpy ndarray 取出滿(mǎn)足特定條件的某些行實(shí)例

猜你喜歡

最新資訊

相關(guān)推薦

相關(guān)標(biāo)簽