Python 过滤numpy数组中的numpy数组
我有一个np数组,我想删除所有符合条件的元素。我想避免for循环,以加快速度 np数组的形状为[N,2]。我想删除第一个元素等于零的所有子数组Python 过滤numpy数组中的numpy数组,python,arrays,numpy,filter,mask,Python,Arrays,Numpy,Filter,Mask,我有一个np数组,我想删除所有符合条件的元素。我想避免for循环,以加快速度 np数组的形状为[N,2]。我想删除第一个元素等于零的所有子数组 [[1,2], [0,5], # <--- Remove [5,1], [0,3], # <--- Remove [1,1], [0,0], # <--- Remove [5,0], ..... [5,5]] [[1,2], [0,5],#如果arr是您的输入数组,请执行以下操作: arr = arr[arr[:,0
[[1,2],
[0,5], # <--- Remove
[5,1],
[0,3], # <--- Remove
[1,1],
[0,0], # <--- Remove
[5,0],
.....
[5,5]]
[[1,2],
[0,5],#如果arr
是您的输入数组,请执行以下操作:
arr = arr[arr[:,0] !=0]
这里的arr[:,0]
对应于第一个元素/列或每个子数组。然后检查arr[:,0]!=0
返回第一个元素不等于零的索引。然后,这些索引将自动传递到arr
,这将提供您所需的内容。如果arr
是您的输入数组,请执行以下操作:
arr = arr[arr[:,0] !=0]
这里的arr[:,0]
对应于第一个元素/列或每个子数组。然后检查arr[:,0]!=0
它返回第一个元素不等于零的索引。然后这些索引会自动传递到arr
,这会提供您想要的内容。所以您有一个包含NumPy数组的NumPy数组?或者您有一个二维数组吗?数组[~(array[:,0]==0)]
这样就可以了。这是一个具有形状(N,2)和整数数据类型的数组,还是一个具有对象数据类型的形状(N,)数组?所以你有一个包含NumPy数组的NumPy数组?还是一个二维数组?数组[~(array[:,0]==0)]
这样就可以了。这是一个具有形状(N,2)和整数数据类型的数组,还是一个具有形状(N,)使用对象数据类型?