根据条件删除熊猫群组

Question

我有一个熊猫数据框与几个群组，我想排除某些条件（特定列）未满足的群组。例如。在这里删除组B，因为它们在列“crit1”中有一个非数字值。

我可以删除基于条件df.loc[:, (df >< 0).any(axis=0)]的特定列，但它不会删除整个组。

不知何故，我不能做出下一步，并将其应用于整个组。

name crit1 crit2 
A  0.3  4 
A  0.7  6 
B  inf  4 
B  0.4  3 

所以结果后，这个过滤（只允许浮动）应为：

A  0.3  4 
A  0.7  6 

Answer 1

您可以使用groupby和filter，你给你可以检查是否存在一组np.inf的例子， filter的条件：

import pandas as pd 
import numpy as np 
df.groupby('name').filter(lambda g: (g != np.inf).all().all()) 
# name crit1 crit2 
# 0 A  0.3  4 
# 1 A  0.7  6 

如果谓词仅适用于一列，可以通过访问列，例如：

df.groupby('name').filter(lambda g: (g.crit1 != np.inf).all()) 
# name crit1 crit2 
# 0 A  0.3  4 
# 1 A  0.7  6