How do I create a new column from the output of pandas groupby().sum()?

by

You want to use transform this will return a Series with the index aligned to the df so you can then add it as a new column:

In [74]:

df = pd.DataFrame({'Date': ['2015-05-08', '2015-05-07', '2015-05-06', '2015-05-05', '2015-05-08', '2015-05-07', '2015-05-06', '2015-05-05'], 'Sym': ['aapl', 'aapl', 'aapl', 'aapl', 'aaww', 'aaww', 'aaww', 'aaww'], 'Data2': [11, 8, 10, 15, 110, 60, 100, 40],'Data3': [5, 8, 6, 1, 50, 100, 60, 120]})
​
df['Data4'] = df['Data3'].groupby(df['Date']).transform('sum')
df
Out[74]:
   Data2  Data3        Date   Sym  Data4
0     11      5  2015-05-08  aapl     55
1      8      8  2015-05-07  aapl    108
2     10      6  2015-05-06  aapl     66
3     15      1  2015-05-05  aapl    121
4    110     50  2015-05-08  aaww     55
5     60    100  2015-05-07  aaww    108
6    100     60  2015-05-06  aaww     66
7     40    120  2015-05-05  aaww    121

More Related Contents:

  1. How can I pivot a dataframe?
  2. Get statistics for each group (such as count, mean, etc) using pandas GroupBy?
  3. GroupBy pandas DataFrame and select most common value
  4. Pandas GroupBy.apply method duplicates first group
  5. pandas GroupBy columns with NaN (missing) values
  6. Count unique values per groups with Pandas [duplicate]
  7. How to access pandas groupby dataframe by key
  8. group by pandas dataframe and select latest in each group
  9. Pandas GroupBy and select rows with the minimum value in a specific column
  10. why does pandas rolling use single dimension ndarray
  11. Group dataframe and get sum AND count?
  12. Sample each group after pandas groupby
  13. Pandas groupby with categories with redundant nan
  14. Is there an “ungroup by” operation opposite to .groupby in pandas?
  15. get first and last values in a groupby
  16. Pandas groupby and aggregation output should include all the original columns (including the ones not aggregated on)
  17. Get the row corresponding to the max in pandas GroupBy [duplicate]
  18. How to get number of groups in a groupby object in pandas?
  19. Pandas: filling missing values by mean in each group
  20. How to groupby consecutive values in pandas DataFrame
  21. Multiple aggregations of the same column using pandas GroupBy.agg()
  22. Pandas ‘count(distinct)’ equivalent
  23. Pandas Groupby Range of Values
  24. Count unique values using pandas groupby
  25. pandas groupby, then sort within groups
  26. group by in group by and average
  27. pandas – find first occurrence
  28. Pivot Tables of Counts in Pandas DataFrame
  29. Bar graph from dataframe groupby
  30. pandas group by and assign a group id then ungroup

Leave a Comment