熊猫：保持前三行包含每个唯一值的值

Question

我有一个数据框，试图将其限制为3次出现唯一值。 我的桌子是这样的：

A B C

1个a
1 b c
1 b a
1 c a
2 c a
2 a b
2 e f
3 e

我想过滤A列。结果表将是

A B C

1个a
1 b c
1 b a
2 c a
2 a b
2 e f
3 e

我现在让代码在iterrows（）循环中工作，但是它有点慢，所以我想知道是否有更好的方法可以做到这一点。