Skip to content Skip to sidebar Skip to footer

Filter Dataframe After Groupby And Nunique In Pandas

i tried df.groupby('item')['variable'].nunique() and it returns a unique count of every item object. i want to filter to only return the count of 'variable' > 3 conditional on

Solution 1:

When you want the groupby to be mapped to every row of the input, think about transform:

df = df[df.groupby("item")["variable"].transform('nunique') > 3]

Post a Comment for "Filter Dataframe After Groupby And Nunique In Pandas"