Python – rolling functions for GroupBy object

by

For the Googlers who come upon this old question:

Regarding @kekert’s comment on @Garrett’s answer to use the new

df.groupby('id')['x'].rolling(2).mean()

rather than the now-deprecated 

df.groupby('id')['x'].apply(pd.rolling_mean, 2, min_periods=1)

curiously, it seems that the new .rolling().mean() approach returns a multi-indexed series, indexed by the group_by column first and then the index. Whereas, the old approach would simply return a series indexed singularly by the original df index, which perhaps makes less sense, but made it very convenient for adding that series as a new column into the original dataframe.

So I think I’ve figured out a solution that uses the new rolling() method and still works the same:

df.groupby('id')['x'].rolling(2).mean().reset_index(0,drop=True)

which should give you the series

0    0.0
1    0.5
2    1.5
3    3.0
4    3.5
5    4.5

which you can add as a column:

df['x'] = df.groupby('id')['x'].rolling(2).mean().reset_index(0,drop=True)

More Related Contents:

  1. How can I pivot a dataframe?
  2. How to group dataframe rows into list in pandas groupby
  3. Concatenate strings from several rows using Pandas groupby
  4. Groupby value counts on the dataframe pandas
  5. Split pandas dataframe based on groupby
  6. GroupBy pandas DataFrame and select most common value
  7. Converting a Pandas GroupBy output from Series to DataFrame
  8. Count unique values per groups with Pandas [duplicate]
  9. Pandas groupby cumulative sum
  10. Python Pandas max value in a group as a new column
  11. pandas dataframe groupby datetime month
  12. How to loop over grouped Pandas dataframe?
  13. How to use groupby to concatenate strings in python pandas?
  14. How to move pandas data from index to column after multiple groupby
  15. Pandas GroupBy and select rows with the minimum value in a specific column
  16. Use pandas.shift() within a group
  17. Pandas Groupby and Sum Only One Column
  18. Python Pandas Conditional Sum with Groupby
  19. Identifying statistical outliers with pandas: groupby and reduce rows into different dataframe
  20. Sample each group after pandas groupby
  21. Pandas groupby with categories with redundant nan
  22. pandas groupby where you get the max of one column and the min of another column
  23. Bar graph from dataframe groupby
  24. pandas group by and assign a group id then ungroup
  25. Is there an “ungroup by” operation opposite to .groupby in pandas?
  26. get first and last values in a groupby
  27. Pandas groupby and aggregation output should include all the original columns (including the ones not aggregated on)
  28. Get the row corresponding to the max in pandas GroupBy [duplicate]
  29. How to get number of groups in a groupby object in pandas?
  30. When is it appropriate to use df.value_counts() vs df.groupby(‘…’).count()?

Leave a Comment