熊猫：使用groupby和函数过滤DataFrame

熊猫：使用groupby和函数过滤DataFrame

len

我认为您要考虑的不是每个长度，而是每个组中Name的唯一值的数量。使用

nunique()

，并检查此简洁的配方以过滤组。

df[df.groupby('ID').Name.transform(lambda x: x.nunique() == 1).astype('bool')]

如果您升级到熊猫0.12，则可以

filter

在组上使用新方法，这将使其更加简洁明了。

df.groupby('ID').filter(lambda x: x.Name.nunique() == 1)

一句话：当然，有时候您确实想知道小组的

size

人数

len

，但是我发现这是一个比更为安全的选择，在某些情况下，这对我来说很麻烦。