{ "componentChunkName": "component---src-templates-blog-post-js", "path": "/pandas-groupby-method/", "result": {"data":{"markdownRemark":{"html":"
\n\n
\n

We can use the groupby method to split a DataFrame into multiple groups/segments based on the unique values in a particular column.

\n

This concept is best illustrated with the help of an example. So, let’s read the budget.xlsx file into a DataFrame:

\n
import pandas as pd\n\ndata = pd.read_excel("budget.xlsx")\n\ndata
\n

Output:

\n

\n \n \n \n \n

\n

Let’s say we want to create multiple segments of our dataset based on the unique values in the Zone column. (You can also choose to segment the dataset based on the LTL Flag or the City column. The decision really depends on your requirements.)

\n

We can check the unique values and their counts in the Zone column using the value_counts() method.

\n
data["Zone"].value_counts()
\n

Output:

\n

\n \n \n \n \n

\n

We can see that there are 4 unique segments (East, West, North, South) that correspond to each unique value in the Zone column.

\n

groups()

\n

We can see the segments we have created using the groups() method.

\n
zones.groups()\n\n{'East': [1, 9], 'North': [0, 2, 6, 8, 10], 'South': [3, 7], 'West': [4, 5]}
\n

Calling the groups method on the zones DataFrameGroupBy object will get us a python dictionary. The keys in the dictionary represent each created segment and the values represent all of the rows that fall under that segment.

\n

get_group()

\n

We can retrieve a segment from our zones DataFrameGroupBy object using the get_group() method, which receives the name of the segment as its argument.

\n
east = zones.get_group("East")\n\neast
\n

This will pull out all of the rows where the value in the Zone column is East.

\n

Output:

\n

\n \n \n \n \n

\n

agg()

\n

We call the agg() method directly on the zones DataFrameGroupBy object.

\n
zones.agg({"Revenue" : ["size", "sum", "mean"], "Profit" :"sum"})
\n

We can perform different aggregations in different columns.

\n

Output:

\n

\n \n \n \n \n

\n","frontmatter":{"title":"Pandas groupby() Method","date":"2021-08-12"}}},"pageContext":{"slug":"/pandas-groupby-method/","prev":{"fields":{"slug":"/pandas-series-filtering-with-string-methods/"},"frontmatter":{"modules":null}},"next":{"fields":{"slug":"/join-pandas-dataframes-with-the-join-method/"},"frontmatter":{"modules":null}}}}, "staticQueryHashes": ["3159585216"]}