python - Numpy 删除符合条件的多行

Question

我有一个具有以下结构的 numpy 数组

sb = np.genfromtxt(open('HomePage/TodayList.txt', 'rb'),
                   delimiter=',', skiprows=0,
                   dtype=[('DataBase', np.str_, 16), ('Mode', np.str_, 16),
                          ('SMB', np.str_, 16),('Desc', np.str_, 128), 
                          ('Res', np.str_, 16), ('RightCnt', np.float64), 
                          ('PercentCnt', np.float64), ('ModelType', np.float64)])

第六栏'PercentCnt'可以通过名称 'PercentCnt' 访问包含从 0 到 50 的数字第七栏'ModelType'包含从 0 到 5 的数字，因此我需要删除或删除符合这些条件的数组行 'PercentCnt'<50和'ModelType'<2 .

Answer 1

条件

sb['PercentCnt'] >= 50

是在此列中保留内容的条件，条件

sb['ModelType'] >= 2

其他列也相同。

您可以将这些与 np.logical_and 结合起来:

keep = np.logical_and(sb['PercentCnt'] >= 50, sb['ModelType'] >= 2)

最后，只需保留您想要保留的行即可:

sb[keep]