numpy ndarray 取出满足特定条件的某些行实例
在进行物体检测的groundtruthboxesannotations包围框坐标数据整理时，需要实现这样的功能：
numpy里面，对于N*4的数组，要实现对于每一行，如果第3列和第1列数值相等或者第2列和第0列数值相等，就删除这一行，要返回保留下来的numpy数组shapeM*4
对于numpy数组的操作要尽量避免for循环，因为numpy数组支持布尔索引。
importnumpyasnp 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)) '''
numpyvstack所输入的参数必须是list或者tuple的iterable对象，在竖直方向上进行数组拼接
其中list或者tuple中的每个元素是numpy.ndarray类型
它们必须具有相同的列数，拼接完成后行数增加
numpyhstack在水平方向上进行数组拼接
进行拼接的数组必须具有相同的行数，拼接完成后列数增加
''' print(b.shape,b) out=b[b[:,3]!=b[:,1]] out2=out[out[:,2]!=out[:,0]] print(out2.shape,out2) ''' (5,4) [[1015] [6991] [9165] [2881] [0858]] (3,4) [[6991] [9165] [2881]] ''' 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)
以上这篇numpyndarray取出满足特定条件的某些行实例就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持毛票票。
声明：本文内容来源于网络，版权归原作者所有，内容由互联网用户自发贡献自行上传，本网站不拥有所有权，未作人工编辑处理，也不承担相关法律责任。如果您发现有涉嫌版权的内容，欢迎发送邮件至：czq8825#qq.com（发邮件时，请将#更换为@）进行举报，并提供相关证据，一经查实，本站将立刻删除涉嫌侵权内容。