Python 按函数筛选numpy数组

我有一个2D numpy数组，格式如下：

array([[0, 4],
       [1, 5],
       [2, 6]])

我想筛选出第一个值大于1的行，但找不到numpy函数

我知道我可以使用

过滤器
：

np.array(list(filter((lambda x: x[0] <= 1), my_arr)))

np.array（list（filter）（（lambda x:x[0]）没有
numpy
接口可以有效地使用函数执行此操作。但是，在这种特殊情况下，您只需要以下内容：

>>> import numpy as np
>>> arr = np.array([[0, 4],
...        [1, 5],
...        [2, 6]])
>>> arr[arr[:,0] <= 1]
array([[0, 4],
       [1, 5]])

>>将numpy作为np导入
>>>arr=np.array（[[0,4]，
...        [1, 5],
...        [2, 6]])
>>>arr[arr[：，0]顺便说一句，您通常应该更喜欢列表理解而不是映射/筛选。
import numpy as np

arr=np.array([[0, 4],
       [1, 5],
       [2, 6]])

newarr=[]

for fetch in arr:
    if(fetch[0]>1):
        newarr.append(fetch)
        
print(newarr)