python - 在 Pandas 数据框的每一组中保留公共(public)行

Question

给定以下 Pandas 数据框:

  | a  b
--+-----
0 | 1  A 
1 | 2  A 
2 | 3  A 
3 | 4  A 
4 | 1  B 
5 | 2  B 
6 | 3  B 
7 | 1  C 
8 | 3  C 
9 | 4  C 

如果按列分组 b我想执行一个操作，只保留它们具有列 a 的行共同点。结果将是以下数据框:

  | a  b
--+-----
0 | 1  A 
2 | 3  A 
4 | 1  B 
6 | 3  B 
7 | 1  C 
8 | 3  C 

是否有一些内置方法可以做到这一点？

Answer 1

你可以试试 pivot_table 与 dropna 然后在这里过滤使用 sreries.isin :

s = df.pivot_table(index='a',columns='b',aggfunc=len).dropna().index
df[df['a'].isin(s)]

与 crosstab 类似:

s = pd.crosstab(df['a'],df['b'])
df[df['a'].isin(s[s.all(axis=1)].index)]

   a  b
0  1  A
2  3  A
4  1  B
6  3  B
7  1  C
8  3  C