python - 选择所有年份都有观察值的行 python

Question

我有大学学生的数据集，我想保留那些没有辍学的学生的观察结果，即多年来对所有人的观察结果例如:

Name        Year
Jacop       2010
Jacop       2011
Jacop       2012
Jacop       2013
Nina        2008
Nina        2009
Nina        2010

我需要一些东西，比如按名称计算值，如果它小于 4，则删除。怎么做？

Answer 1

我认为你需要filter :

df = df.groupby('Name').filter(lambda x: len(x) >= 4)
print (df)
    Name  Year
0  Jacop  2010
1  Jacop  2011
2  Jacop  2012
3  Jacop  2013

另一个解决方案 transform和 boolean indexing :

df = df[df.groupby('Name')['Name'].transform('size') >= 4]
print (df)
    Name  Year
0  Jacop  2010
1  Jacop  2011
2  Jacop  2012
3  Jacop  2013