获取与numpy中的条件匹配的行的行号

获取与numpy中的条件匹配的行的行号

使用

np.where

返回指数：

In [79]:np.where(a[:,1]==2)Out[79]:(array([0, 3], dtype=int64),)