Pandas: Split a given dataframe into groups and create a new column with count from GroupBy
17. Creating New Column with Group Count
Write a Pandas program to split a given dataframe into groups and create a new column with count from GroupBy.
Test Data:
book_name book_type book_id 0 Book1 Math 1 1 Book2 Physics 2 2 Book3 Computer 3 3 Book4 Science 4 4 Book1 Math 1 5 Book2 Physics 2 6 Book3 Computer 3 7 Book5 English 5
Sample Solution:
Python Code :
Sample Output:
Original Orders DataFrame: book_name book_type book_id 0 Book1 Math 1 1 Book2 Physics 2 2 Book3 Computer 3 3 Book4 Science 4 4 Book1 Math 1 5 Book2 Physics 2 6 Book3 Computer 3 7 Book5 English 5 New column with count from groupby: book_name book_type count 0 Book1 Math 2 1 Book2 Physics 2 2 Book3 Computer 2 3 Book4 Science 1 4 Book5 English 1
For more Practice: Solve these Related Problems:
- Write a Pandas program to group a dataframe and then add a new column that represents the count of rows in each group.
- Write a Pandas program to split a dataframe into groups and then assign each row a group count based on its group.
- Write a Pandas program to group the data and then merge the group sizes as a new column into the original dataframe.
- Write a Pandas program to calculate the count of each group and then join this count back to the dataframe as an additional column.
Go to:
Previous: Write a Pandas program to split a given dataframe into groups and list all the keys from the GroupBy object.
Next: Write a Pandas program to split a given dataframe into groups with bin counts.
Python Code Editor:
Have another way to solve this solution? Contribute your code (and comments) through Disqus.
What is the difficulty level of this exercise?
Test your Programming skills with w3resource's quiz.