python - Pandas 数据框过滤

Question

给定一个包含两列的数据框:用户和代码，我如何过滤掉没有至少 x 个具有给定代码的条目的用户条目？

例如我想过滤掉所有用户，当他们没有至少 5 次出现给定类型时:

User    Type
A       Alpha
A       Alpha
A       Alpha
A       Alpha
A       Alpha
A       Beta
A       Beta
A       Beta
B       Alpha
B       Alpha
B       Alpha
B       Alpha
B       Alpha

在这里我想过滤掉(删除)带有 Beta 代码的 4x A(这里只有 4 次)，同时保留其他所有内容。

谢谢!

Answer 1

你可以groupby 'User' 和 'Type' 和 filter:

In [91]:
df.groupby(['User', 'Type']).filter(lambda x: len(x) > 4)

Out[91]:
   User   Type
0     A  Alpha
1     A  Alpha
2     A  Alpha
3     A  Alpha
4     A  Alpha
8     B  Alpha
9     B  Alpha
10    B  Alpha
11    B  Alpha
12    B  Alpha