Group by and Filter Groups in Pandas

Last update on December 21 2024 09:22:53 (UTC/GMT +8 hours)

Pandas Advanced Grouping and Aggregation: Exercise-4 with Solution

Group by and Filter Groups:
Write a Pandas program that implements the technique of grouping and filtering groups to refine your data analysis and insights.

Sample Solution:

Python Code :

import pandas as pd
# Sample DataFrame
data = {'Category': ['A', 'A', 'B', 'B', 'C', 'C'],
        'Value': [1, 2, 3, 4, 5, 6]}
df = pd.DataFrame(data)
print("Sample DataFrame:")
print(df)
# Group by 'Category'
grouped = df.groupby('Category')
# Filter groups where the sum of 'Value' > 5
print("\nFilter groups where the sum of 'Value' > 5")
filtered = grouped.filter(lambda x: x['Value'].sum() > 5)

print(filtered)

Output:

Sample DataFrame:
  Category  Value
0        A      1
1        A      2
2        B      3
3        B      4
4        C      5
5        C      6

Filter groups where the sum of 'Value' > 5
  Category  Value
2        B      3
3        B      4
4        C      5
5        C      6

Explanation:

• Import pandas.
• Create a sample DataFrame.
• Group by 'Category'.
• Filter groups where the sum of 'Value' > 5.
• Print the filtered DataFrame.

Python Code Editor:

Have another way to solve this solution? Contribute your code (and comments) through Disqus.

Previous: Use Custom Aggregation Functions in Pandas GroupBy.
Next: Group by and Apply function to Groups in Pandas.