如何過濾成員多於1個的組中的所有行？

Question

我有一個類似的問題，因為這一個 。

我有一個這樣的數據框：

import pandas as pd

df = pd.DataFrame({'A': list(range(7)),
                   'B': ['a', 'b', 'a', 'c', 'c', 'b', 'b'],
                   'C': ['x', 'x', 'x', 'z', 'z', 'y', 'x']}
                  )

   A  B  C
0  0  a  x
1  1  b  x
2  2  a  x
3  3  c  z
4  4  c  z
5  5  b  y
6  6  b  x

我想groupby列B和C ，然后選擇從所有的行df有一組尺寸大於1。

我想要的結果是

   A  B  C
0  0  a  x
1  1  b  x
2  2  a  x
3  3  c  z
4  4  c  z
6  6  b  x

所以我可以做

gs_bool = df.groupby(['B', 'C']).size() > 1

這使

B  C
a  x     True
b  x     True
   y    False
c  z     True
dtype: bool

現在如何將其反饋給df ？

Answer 1

您真的很親密-需要GroupBy.transform ：

gs_bool = df.groupby(['B', 'C'])['B'].transform('size') > 1
print (gs_bool)
0     True
1     True
2     True
3     True
4     True
5    False
6     True
Name: B, dtype: bool

---

df = df[gs_bool]
print (df)
   A  B  C
0  0  a  x
1  1  b  x
2  2  a  x
3  3  c  z
4  4  c  z
6  6  b  x

Answer 2

IIUC：

In [38]: df.groupby(['B','C']).filter(lambda x: len(x) > 1)
Out[38]:
   A  B  C
0  0  a  x
1  1  b  x
2  2  a  x
3  3  c  z
4  4  c  z
6  6  b  x