Python: Removing Rows on Count condition
This is one way using pd.Series.value_counts
.
counts = df['city'].value_counts()res = df[~df['city'].isin(counts[counts < 5].index)]
counts
is a pd.Series
object. counts < 5
returns a Boolean series. We filter the counts series by the Boolean counts < 5
series (that's what the square brackets achieve). We then take the index of the resultant series to find the cities with < 5 counts. ~
is the negation operator.
Remember a series is a mapping between index and value. The index of a series does not necessarily contain unique values, but this is guaranteed with the output of value_counts
.
I think you're looking for value_counts()
# Import the great and powerful pandasimport pandas as pd# Create some example datadf = pd.DataFrame({ 'city': ['NYC', 'NYC', 'SYD', 'NYC', 'SEL', 'NYC', 'NYC']})# Get the count of each valuevalue_counts = df['city'].value_counts()# Select the values where the count is less than 3 (or 5 if you like)to_remove = value_counts[value_counts <= 3].index# Keep rows where the city column is not in to_removedf = df[~df.city.isin(to_remove)]