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

Question

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

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

标签： pythonnumpy

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]])