numpy ndarray 取出满足特定条件的某些行实例
更新时间：2019年12月05日 13:42:14 作者：WYXHAHAHA123
今天小编就为大家分享一篇numpy ndarray 取出满足特定条件的某些行实例，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
在进行物体检测的ground truth boxes annotations包围框坐标数据整理时，需要实现这样的功能：
numpy里面，对于N*4的数组，要实现对于每一行，如果第3列和第1列数值相等或者第2列和第0列数值相等，就删除这一行，要返回保留下来的numpy数组 shape M*4
对于numpy数组的操作要尽量避免for循环，因为numpy数组支持布尔索引。
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 所输入的参数必须是list或者tuple的iterable对象，在竖直方向上进行数组拼接
其中list或者tuple中的每个元素是numpy.ndarray类型
它们必须具有相同的列数，拼接完成后行数增加
numpy hstack 在水平方向上进行数组拼接
进行拼接的数组必须具有相同的行数，拼接完成后列数增加
''' 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 取出满足特定条件的某些行实例就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持脚本之家。
相关文章
今天给大家分享的是如何使用Django实现简单分页功能的方法，主要依托于Django提供的一个新的类Paginator来实现的2018-07-07
这篇文章主要介绍了Python中with语句的用法,with语句的使用是Python学习过程当中的基础知识,本文来自于IBM官方技术文档,需要的朋友可以参考下2015-04-04
在机器视觉领域中,卷积神经网络算法作为一种新兴算法出现,在图像识别领域中,卷积神经网络能够较好的实现图像的分类效果,而且其位移和形变具有较高的容忍能力2021-11-11
对numpy的array和python中自带的list之间相互转化详解
下面小编就为大家分享一篇对numpy的array和python中自带的list之间相互转化详解，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2018-04-04
这篇文章主要介绍了Python 函数返回值的示例代码，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧2019-03-03
不知道大家有没有发现,我们在网页移动鼠标时,不同的网页会有不同的鼠标移动特效,通过移动鼠标,会形成类似蜘蛛网等等的特效,本文将用PyQt5实现这一特效,需要的可以参考一下2024-07-07
这篇文章主要介绍了Python基于字典实现switch case函数调用,文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下2020-07-07
在本篇内容中小编给大家分享的是关于python是否属于脚本语言的相关内容知识点，有兴趣的朋友们可以参考下。2020-06-06
这篇文章主要为大家详细介绍了如何使用Python实现批量将文件名称符合要求的文件自动复制到新文件夹,文中的示例代码讲解详细,有需要的小伙伴可以参考下2023-10-10
Python Pandas:DataFrame一列切分成多列、分隔符切分选字段方式
这篇文章主要介绍了Python Pandas:DataFrame一列切分成多列、分隔符切分选字段方式,具有很好的参考价值,希望对大家有所帮助,如有错误或未考虑完全的地方,望不吝赐教2023-09-09
最新评论