pandas - 如何按两行比较值

Question

如何选择列 [nbr] 包含不同值的行。

可能某些 [st_id] 只有一条记录。

   st_id       trc      dir  nbr    
   88900     4009114     1   2  
   88900     4009114     2   2    
   88000     4009115     1   2    
   88000     4009115     2   2    
   88300     4009113     1   3   
   88300     4009113     2   2   
   88400     4009110     1   4  
   88500     5120012     1   1
   88500     5120013     2   2
   88600     1270081     1   3

结果：

   st_id       trc      dir  nbr    
   88300     4009113     1   3   
   88300     4009113     2   2   
   88500     5120012     1   1
   88500     5120013     2   2

标签： pandaspandas-groupby

Answer 1

您可以首先使用drop_duplicates为每对获取一行st_id, nbr，然后groupby仅st_id用于查找存在多个实例的实例nbr：

(df
 .drop_duplicates(["st_id", "nbr"])
 .groupby(["st_id"])
 .filter(lambda x: x.nbr.nunique() > 1)
)

   st_id      trc  dir  nbr
0  88300  4009114    1    2
4  88300  4009113    1    3
7  88500  5120012    1    1
8  88500  5120013    2    2