python – Numpy – 删除负值的数据行

发布时间：2020-12-20 11:50:49 所属栏目：Python 来源：网络整理

导读：我正在从AUTO数字延续包中获取输出,并且需要过滤掉具有负值变量的结果,因为它们是非物理的.所以,如果我有,例如： a = np.array([[0,1,2,3,4],[-1,-0.5,0.5,1],[-3,-4,-5,0.1,0.2]]) 我想留下： barray([[ 3.,4. ],[ 0.5,1. ],[ 0.1,0.2]]) 但当我尝试numpy.w

我正在从AUTO数字延续包中获取输出,并且需要过滤掉具有负值变量的结果,因为它们是非物理的.所以,如果我有,例如：

>>>   a = np.array([[0,1,2,3,4],[-1,-0.5,0.5,1],[-3,-4,-5,0.1,0.2]])

我想留下：

>>> b
array([[ 3.,4. ],[ 0.5,1. ],[ 0.1,0.2]])

但当我尝试numpy.where得到：

>>> b = a[:,(np.where(a[1]>=0) and np.where(a[2]>=0))]
>>> b
array([[[ 3.,4. ]],[[ 0.5,1. ]],[[ 0.1,0.2]]])
>>> b.shape
(3,2)

也就是说,它会向阵列添加另一个不需要的轴.我究竟做错了什么？

解决方法

假设您要做的就是删除具有一个或多个负值的列,您可以这样做：

a = np.array([[0,0.2]])
b = a[:,a.min(axis=0)>=0]

（编辑：李大同）

【声明】本站内容均来自网络，其相关言论仅代表作者个人观点，不代表本站立场。若无意侵犯到您的权利，请及时与联系站长删除相关内容!