根據條件刪除熊貓羣組

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