Skip to content

Instantly share code, notes, and snippets.

@icexelloss

icexelloss/substract_mean.py

Last active March 1, 2018 15:25
Show Gist options
  • Save icexelloss/f3572c728e402b4cff21a8581e5d2db1 to your computer and use it in GitHub Desktop.
Save icexelloss/f3572c728e402b4cff21a8581e5d2db1 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
substract_mean.py
@pandas_udf(df.schema, PandasUDFType.GROUPED_MAP)
# Input/output are both a pandas.DataFrame
def subtract_mean(pdf):
return pdf.assign(v=pdf.v - pdf.v.mean())
df.groupby('id').apply(subtract_mean)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment