如何删除基于单列的numpy数组中的行？

Question

我有一个浮点数的2d numpy数组，并且我想删除该行的第三列所包含的值小于x所有行。

例如。 具有阈值2 [[3,4,5],[3,3,8],[4,2,1],[1,2,1]] ，输出[[3,4,5],[3,3,8]] 。

Answer 1

试试这个：

>>> import numpy as np
>>> x=np.array([[3,4,5],[3,3,8],[4,2,1],[1,2,1]])
>>> x=x[x[:,2]>=2]
>>> x
array([[3, 4, 5],
       [3, 3, 8]])

Answer 2

您可以使用列表理解：

import numpy as np

arr = np.array([[3,4,5],[3,3,8],[4,2,1],[1,2,1]])

threshold = 2

arr = np.array([row for row in arr if row[2] >= threshold])

print(arr)

输出：

[[3 4 5]
 [3 3 8]]

另外，您可以使用filter ：

np.array([*filter(lambda r : r[2] >= threshold, arr)])

Answer 3

尝试这个：

import numpy as np

array = np.array([[3,4,5],[3,3,8],[4,2,1],[1,2,1]])
array = np.array([x for x in array if x[2] >  2])
print (array)