pandas add column to groupby dataframe

by

Use transform to add a column back to the orig df from a groupby aggregation, transform returns a Series with its index aligned to the orig df:

In [123]:
g = df.groupby('c')['type'].value_counts().reset_index(name="t")
g['size'] = df.groupby('c')['type'].transform('size')
g

Out[123]:
   c type  t  size
0  1    m  1     3
1  1    n  1     3
2  1    o  1     3
3  2    m  2     4
4  2    n  2     4

More Related Contents:

  1. how to calculation cost time [closed]
  2. pandas: filter rows of DataFrame with operator chaining
  3. Pandas groupby: How to get a union of strings
  4. Reshape wide to long in pandas
  5. How do I create test and train samples from one dataframe with pandas?
  6. pandas groupby sort within groups
  7. Applying function with multiple arguments to create a new pandas column
  8. pandas groupby, then sort within groups
  9. Converting between datetime and Pandas Timestamp objects
  10. How to query MultiIndex index columns values in pandas
  11. Add multiple empty columns to pandas DataFrame
  12. Shift NaNs to the end of their respective rows
  13. multi index plotting
  14. No numeric types to aggregate – change in groupby() behaviour?
  15. Pandas how can ‘replace’ work after ‘loc’?
  16. Pandas Dataframe: split column into multiple columns, right-align inconsistent cell entries
  17. Pandas in AWS lambda gives numpy error
  18. Pandas: Reading Excel with merged cells
  19. Pandas groupby with categories with redundant nan
  20. Python numpy: cannot convert datetime64[ns] to datetime64[D] (to use with Numba)
  21. Slicing multiple ranges of columns in Pandas, by list of names
  22. pandas convert from datetime to integer timestamp
  23. mean from pandas and numpy differ
  24. Python: Datetime to season
  25. Replace NaN with empty list in a pandas dataframe
  26. How to set the xlim and ylim of a FacetGrid
  27. Problem in combining bar plot and line plot
  28. When is it appropriate to use df.value_counts() vs df.groupby(‘…’).count()?
  29. Plot all pandas dataframe columns separately
  30. Python pandas equivalent for replace

Leave a Comment