Pandas
keyboard_arrow_down 655 guides
chevron_leftData Aggregation Cookbook
Applying a function to multiple columns in groupsCalculating percentiles of a DataFrameCalculating the percentage of each value in each groupComputing descriptive statistics of each groupDifference between a group's count and sizeDifference between methods apply and transform for groupbyGetting cumulative sum of each groupGetting descriptive statistics of DataFrameGetting multiple aggregates of a column after groupingGetting n rows with smallest column value in each groupGetting number of distinct rows in each groupGetting size of each groupGetting specific group after groupbyGetting the first row of each groupGetting the last row of each groupGetting the top n rows with largest column value in each groupGetting unique values of each groupGrouping by multiple columnsGrouping without turning group column into indexMerging rows within a group togetherNaming columns after aggregationSorting values within groups
check_circle
Mark as learned thumb_up
4
thumb_down
0
chat_bubble_outline
0
Comment auto_stories Bi-column layout
settings
Applying a function to multiple columns in groups in Pandas
schedule Aug 12, 2023
Last updated local_offer
Tags Python●Pandas
tocTable of Contents
expand_more Master the mathematics behind data science with 100+ top-tier guides
Start your free 7-days trial now!
Start your free 7-days trial now!
Consider the following DataFrame:
df = pd.DataFrame({"A":[2,3,4],"B":[5,6,7],"group":["a","a","b"]})df
A B group0 2 5 a1 3 6 a2 4 7 b
To group by group
, and then apply a function on multiple columns in each group:
def f(my_df): return pd.Series({"C": my_df["A"].sum() + my_df["B"].sum()})
df.groupby("group").apply(f)
Cgroup a 16b 11
Note the following:
our
f
function is called twice in this case - once for each groupargument for
f
is a DataFrame representing each group
Published by Isshin Inada
Edited by 0 others
Did you find this page useful?
thumb_up
thumb_down
Comment
Citation
Ask a question or leave a feedback...
thumb_up
4
thumb_down
0
chat_bubble_outline
0
settings
Enjoy our search
Hit / to insta-search docs and recipes!