Pandas create new column with count from groupby

by

That’s not a new column, that’s a new DataFrame:

In [11]: df.groupby(["item", "color"]).count()
Out[11]:
             id
item  color
car   black   2
truck blue    1
      red     2

To get the result you want is to use reset_index:

In [12]: df.groupby(["item", "color"])["id"].count().reset_index(name="count")
Out[12]:
    item  color  count
0    car  black      2
1  truck   blue      1
2  truck    red      2

To get a “new column” you could use transform:

In [13]: df.groupby(["item", "color"])["id"].transform("count")
Out[13]:
0    2
1    2
2    2
3    1
4    2
dtype: int64

I recommend reading the split-apply-combine section of the docs.

More Related Contents:

  1. Replacing values represented by 'UN' with NaN
  2. Renaming column names in Pandas
  3. Pandas: convert categories to numbers
  4. What’s the fastest way in Python to calculate cosine similarity given sparse matrix data?
  5. Seaborn Barplot – Displaying Values
  6. How can I map True/False to 1/0 in a Pandas DataFrame?
  7. multiprocessing in python – sharing large object (e.g. pandas dataframe) between multiple processes
  8. Is there a simple way to change a column of yes/no to 1/0 in a Pandas dataframe?
  9. Changing multiple column names but not all of them – Pandas Python
  10. python pandas flatten a dataframe to a list
  11. creating multiple excel worksheets using data in a pandas dataframe
  12. Get all combinations of elements from two lists?
  13. How to create a dictionary of two pandas DataFrame columns
  14. Can You Consistently Keep Track of Column Labels Using Sklearn’s Transformer API?
  15. AttributeError: ‘DataFrame’ object has no attribute ‘ix’
  16. What is dtype(‘O’), in pandas?
  17. Counting the Frequency of words in a pandas data frame
  18. UnicodeDecodeError: ‘utf-8’ codec can’t decode byte 0x96 in position 35: invalid start byte
  19. Co-occurrence matrix from nested list of words
  20. How to increment a row count in groupby in DataFrame
  21. Which is the fastest way to extract day, month and year from a given date?
  22. Identifying consecutive NaNs with Pandas
  23. Memory leak using pandas dataframe
  24. How to merge a Series and DataFrame
  25. dataframe to dict such that one column is the key and the other is the value [duplicate]
  26. Custom sorting with Pandas
  27. How to compute jaccard similarity from a pandas dataframe
  28. Pandas: remove duplicates that exist in any order
  29. Difference between df[x], df[[x]], df[‘x’] , df[[‘x’]] and df.x
  30. Numpy.dtype has the wrong size, try recompiling

Leave a Comment